如何在 python 中的 numpy 數組的子數組中獲取元素

Question

我在讀取 python 中 numpy 數組的子數組條目時遇到了麻煩。 我有這樣的事情：

a = np.array([ [453,254,[1,2,3,4,5]], [743,251,[10,20,30,40,50]], [127,393,[11,22,33,44,55]] ], dtype=object)

我需要計算第二個位置子數組的每一列的平均值，即np.mean([1,10,11]), np.mean([2,20,22])等。

如何獲得[1,10,11], [2,20,22]等子數組？

我嘗試過使用":"和","的不同組合，但我無法弄清楚。 我還將 dtype dtype=object放在 numpy 數組定義中，但沒有任何區別。 提前致謝。

Answer 1

你可以這樣做：

np.mean([x[0] for x in a[:,2]])
np.mean([x[1] for x in a[:,2]])
...

Answer 2

我找到了一個可能的解決方案

np.dstack((a[:,2]))[0][0]