Python NumPy изменить

Question

У меня есть переменная z, хранящая характеристики длины и высоты файлов изображений, где z равно

z = [length, height]

и я хочу изменить это измерение на:

z = [area] where area = length * height

Я попытался использовать функцию numpy reshape следующим образом:

area = z.shape[0] * z.shape[1] #length * height
z = z.reshape(-1) #was trying to reduce to just z = [area]

но мне показалось, что я не правильно использую функцию изменения формы. Кто-нибудь может мне помочь?

Answer 1

Простой пример использования reshape:

import numpy as np

a = np.random.randint(0,10,(10,10))
b = np.reshape(a, (100,))
print(b)

Для вашего случая это будет:

print(a.shape)  # prints (length,height)
b = np.reshape(a, (length * height,))
print(b.shape)  # prints (length * height,)

Для изменения формы на месте вы также можете использовать:

a.shape = ((100,))