如何在没有for循环的情况下在2D numpy数组中设置多个列？

Question

我想设置numpy数组的值如下.但我不想用for-loop.有什么好办法吗？

a = range(4)
a[0] = [11,12,13,14,15,16]
a[1] = [21,22,23,24,25,26]
a[2] = [31,32,33,34,35,36]
a[3] = [41,42,43,44,45,46]

a = np.array(a)

changeIndex = [0,2,4]
for i in range(4):
    a[i][changeIndex] = 0

print a
#array([[ 0, 12,  0, 14,  0, 16],
#       [ 0, 22,  0, 24,  0, 26],
#       [ 0, 32,  0, 34,  0, 36],
#       [ 0, 42,  0, 44,  0, 46]])

Answer 1

您基本上想要访问多个列,您可以通过以下方式执行:

a[:, changeIndex] = 0

记得:

1. 第一个索引选择行,而第二个索引选择列.
2. 您可以使用列表或元组选择多个索引.

更好的风格:

您还可以使用更好的方法来定义多维numpy数组或矩阵:

a = np.array([range(11,17), range(21,27), range(31, 37), range(41,47)])

并且由于其中一条评论,您实际上可以使用它np.arange()来代替range()更快的计算

矩阵:

在处理2D阵列时,请考虑使用Matrix.当您对它们执行操作时,矩阵保持其多维特性,您还可以对它们使用特殊矩阵运算.阅读这里.它们的工作方式类似于数组:

a = np.matrix([range(11,17), range(21,27), range(31, 37), range(41,47)])

供大家参考有点相关的线程是在这里.