numpy的array数组操作

来源:互联网 发布:约拍用什么软件最好 编辑:程序博客网 时间:2024/05/16 01:26

存取

  • 构造一维array
a = np.array([0, 1, 2, 4, 5, 6])

a[5] ==> 4
a[3:5] ==> [2,4] 不包括a[5]
a[:3] ==> [0, 1, 2]表示从a[0]开始到a[5]
a[:-1] ==> [6, 5, 4, 3, 2, 1, 0]表示从后往前数
a[1:-1:2] ==> [1, 4]第一个参数表示开始坐标;第二个参数表示结束坐标,-1代表结尾,但不包含结尾坐标;第三个表示步长
a[::-1] ==> [6, 5, 4, 3, 2, 1, 0]省略开始与结束坐标,步长为-1代表,从尾部到头部颠倒

  • 构造二维array
a = np.array([[0, 1, 2], [3, 4, 5]])

a[:,:2] ==> [[0, 1], [3, 4]]取所有的行,列从第0列开始,到第1列,取两列
a[:,-2:] ==> [[1, 2], [4, 5]]取所有的行,负数代表,列从第最后一列开始,取两列

当需要读取固定某几行时

a = np.array([[0, 1, 2], [3, 4, 5]])index_list = [0, 1]

a[index_list, :] ==> [[0, 1, 2], [3, 4, 5]]

赋值


a = np.array([1, 2])row = np.array([0, 1])column = np.array([0, 1])qq = np.array([[10, 10],[10, 10]])qq[row, column] = aprint qq

输出:[[1, 10], [10, 2]] # row和column数组中的数值分别组成(0,0),(1,1)

原创粉丝点击