Python: numpy数组添加一行或者一列, numpy数组的增删查改

一、numpy数组的增删查改

(一)查和改

   1.通过下标访问

        array[0][0]

        array[0,1]

    2.通过下标修改

        array[0][0] = 1

        array[0,0] = 2

(二)增

    1.增加一列

    (1)通过np.c_[a,b]方式

import numpy as np
a = np.array([[1,2,3],[4,5,6],[7,8,9]]) 
b = np.array([2,5,8])
print(np.c_[a,b])

输出:
[[1 2 3 2]
 [4 5 6 5]
 [7 8 9 8]]

    (2)通过np.insert(a, index, b, axis=1)方式

import numpy as np
a = np.array([[1,2,3],[4,5,6],[7,8,9]]) 
b = np.array([2,5,8])
# 2代表下标,这里代表插入到第三列,axis=1,插入一列,axis=0,插入一行
print(np.insert(a, 2, b, axis=1))
#插入多列
c = np.array([[1,4,7],[2,5,8]]).T
print(np.insert(a, [0,1], c, axis=1))

输出:
[[1 2 2 3]
 [4 5 5 6]
 [7 8 8 9]]

[[1 1 2 2 3]
 [4 4 5 5 6]
 [7 7 8 8 9]]

   (3)通过np.column_stack((a,b))的方式,效果跟第一种一样

      注意,第一种和第三种只能插入到最后一列,而insert的方法可以插入到任意一列

    2.增加一行

    (1)通过np.r_[a,b]方法

import numpy as np
a = np.array([[1,2,3],[4,5,6],[7,8,9]]) 
b = np.array([2,5,8])
print(np.r_[a,[b]])

输出:
[[1 2 3]
 [4 5 6]
 [7 8 9]
 [2 5 8]]

    注意这里a和b的shape必须要一致 ,这跟插入一列不一样

    (2)通过np.row_stack((a,b))方法

import numpy as np
a = np.array([[1,2,3],[4,5,6],[7,8,9]]) 
b = np.array([2,5,8])
print(np.row_stack((a,b)))

输出:
[[1 2 3]
 [4 5 6]
 [7 8 9]
 [2 5 8]]

    这里b和a的一行形状相同即可 

    (3)通过np.insert(a, index, b, axis=0)的方法

import numpy as np
a = np.array([[1,2,3],[4,5,6],[7,8,9]]) 
b = np.array([2,5,8])
print(np.insert(a, 1, b, axis=0))
# 可以插入多行
c = np.array([[1,4,7],[2,5,8]])
print(np.insert(a, [0,1], c, axis=0))

输出:
[[1 2 3]
 [2 5 8]
 [4 5 6]
 [7 8 9]]

[[1 4 7]
 [1 2 3]
 [2 5 8]
 [4 5 6]
 [7 8 9]]

(三)删 

    1.删除一列或一行

import numpy as np
a = np.array([[1,2,3],[4,5,6],[7,8,9]]) 
print(np.delete(a, 1, axis=0))
print(np.delete(a, -1, axis=1))

输出:
[[1 2 3]
 [7 8 9]]
[[1 2]
 [4 5]
 [7 8]]

    2.删除多列或多行 

import numpy as np
a = np.array([[1,2,3],[4,5,6],[7,8,9]]) 
print(np.delete(a, [0,1], axis=0))
print(np.delete(a, [1,2], axis=1))

输出:
[[7 8 9]]
[[1]
 [4]
 [7]]

   注意,当删除多行或多列时时,不能使用倒序下标-1,-2 

二、参考文献 

    【1】https://www.cnblogs.com/MrLJC/p/4823499.html

    【2】https://blog.csdn.net/zzzzlei123123123/article/details/100154501

原创文章 26 获赞 33 访问量 1898

猜你喜欢

转载自blog.csdn.net/qq_40765537/article/details/105869910
今日推荐