我有一个二维数组,其中第一列填充了一些值,其他列为零。我希望能够使用numpy实现与MS Excel相同的操作,即根据第一列进行计算,并将计算结果填充到其他列中。以下是一个最小化工作示例:
import numpy as np
a = np.zeros(20, dtype=np.int8).reshape(4,5)
b = [1, 2, 3, 4]
b = np.array(b)
a[:, 0] = b
# don't change the first column
for column in a[:, 1:]:
a[:, column] = column[0]+1
预期输出:
array([[1, 2, 3, 4, 5],
[2, 3, 4, 5, 6],
[3, 4, 5, 6, 7],
[4, 5, 6, 7, 8]], dtype=int8)
生成的输出内容:
array([[1, 0, 0, 0, 0],
[1, 0, 0, 0, 0],
[1, 0, 0, 0, 0],
[1, 0, 0, 0, 0]], dtype=int8)
任何帮助将不胜感激。
column1
的值。它是列的值还是索引? - hpaulj