python增加数组元素 python数组增加数据

一、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