Как преобразовать тип элемента в массиве numpy из строки в int

Question

У меня есть массив numpy, который выглядит следующим образом:

arr = np.array([['1','2','3','4','5','6']])

. Для наиболее распространенного способа преобразования требуются циклы, подобные этому.

for a in arr:
   for b in a:
      int(b)

Однако я хотел бы нравится конвертировать все элементы в массиве без циклов. Как я мог это сделать?

Answer 1

Вы можете определить тип элементов в присваивании:

arr = numpy.array(['1','2','3','4','5','6'], int)

или, если вы хотите проанализировать тип каждого элемента в инициализированном массиве, вы можете использовать astype() method

arr.astype('int')