Python Double for循环将新行添加到新的Numpy数组

时间:2019-02-14 13:13:34

标签: python python-3.x numpy

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

0 个答案:

没有答案