Блог

Как мне получить длину столбца в ND-массиве?

например, у меня есть массив nD с именем a. когда я печатаю.shape, он возвращает (1,21). Я хочу выполнить цикл for в диапазоне размера столбца массива a . Как мне получить значение

Вы можете получить второе измерение массива как:

 a.shape[1]
  

Вы могли бы использовать np.size(element,dimension) .

В вашем случае:

 a.shape
>> (1, 21)


np.size(a,0)
>> 1

np.size(a,1)
>> 21
  

Использование 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
  

Решение :

Это решается путем добавления простой опции в array,

 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/

это простой пример, и вы можете следовать ему.Кстати, последнее утверждение следует исправить как «print b.shape [0]»