如何使用numpy仅取矩阵/多维数组的第二列之和

Question

我的数组有两列，第一列是字母（作为单个字母字符串），第二列是数字。我想找到这些数字的总和，但是由于字符串而导致错误。

我发现您可以在numpy.sum函数中执行axis = 0来获取列的总和，这可能更接近我的需要。

import numpy as np
str = "aaabcc"
foo = [
    ["a", str.count("a")],
    ["b", str.count("b")],
    ["c", str.count("c")],
    ]
print(np.sum(foo, axis=0)

如果有一种方法可以在不使axis = 0的情况下获得第二列的总和，则该方法同样适用。

Answer 1

您可以尝试：

main.js

输出：

np.array(foo)[:, 1].astype('int').sum()

Answer 2

像这样吗？

sum(i[1] for i in foo)

Out[1]:
    6