Python。我想比较两列元素是否相同,我想从第三列中获得它们的元素。 创建一个新的Numpy数组,并在其中添加所有这3个元素。 现在,它替换了所有元素,我只得到一个向量,而不是一个长矩阵。 (总和可以)
#data is a numpy array
dataTitle = np.array(["A", "B", "SUM"])
c = 0
for i in range(len(data)):
for j in range(len(data)):
if((data[i, 0] == data[j, 0]) and (data[i, 1] == data[j, 1])):
c += data[j, 3]
data2 = np.array([[data[j, 0], data[j, 1], c]])
np.concatenate(dataTitle, data2, axis = 0)
c = 0