python/numpy：如何獲得二維數組列長度？

Question

如何獲得 nD 數組中列的長度？

例如，我有一個名為 a 的 nD 數組。 當我打印 a.shape 時，它​​返回 (1,21)。 我想做一個for循環，在數組a的列大小范圍內。 我如何獲得價值

Answer 1

您可以獲得數組的第二個維度：

a.shape[1]

Answer 2

您可以使用np.size(element,dimension) 。

在你的情況下：

a.shape
>> (1, 21)


np.size(a,0)
>> 1

np.size(a,1)
>> 21

Answer 3

在定義二維數組時使用shape和size效果很好，但是在定義簡單數組時，這些方法不起作用例如：

K = np.array([0,2,0])

K.shape[1]和numpy.size(K,1)

在 python 中產生錯誤：

Traceback (most recent call last):
  File "<ipython-input-46-e09d33390b10>", line 1, in <module>
    K.shape[1]
IndexError: tuple index out of range

解決方案 ：

它通過向數組添加一個簡單的選項來解決，

K = np.array([0,2,0],ndmin = 2)


K.shape[0]
Out[56]: 1

K.shape[1]
Out[57]: 3

np.size(K,0)
Out[58]: 1

np.size(K,1)
Out[59]: 3

更多信息 ：

https://codewithkazem.com/array-shape1-indexerror-tuple-index-out-of-range/

Answer 4

這是一個簡單的例子，你可以照着做。順便說一句，最后一句應該更正為“print b.shape[0]”