Я исхожу из фона MATLAB, и эта простая операция до сих пор кажется чрезвычайно сложной для реализации в Python, согласно ответам в других стеках. Как правило, в большинстве ответов используется цикл for.
Лучшее, что я видел до сих пор, это
import numpy
start_list = [5, 3, 1, 2, 4]
b = list(numpy.array(start_list)**2)
Есть ли более простой способ?
Наиболее читаемым, вероятно, является понимание списка:
start_list = [5, 3, 1, 2, 4]
b = [x**2 for x in start_list]
Если вы функциональный тип, вам понравится map:
b = map(lambda x: x**2, start_list) # wrap with list() in Python3
Примечание. Поскольку у нас уже есть дубликаты для ванильного Python, списки и сопоставления и что я не нашел дубликата для возведения в квадрат массива 1D numpy, я подумал, что оставлю свой первоначальный ответ, в котором используется numpy
numpy.square()Если вы переходите с MATLAB на Python, вы определенно правы, пытаясь использовать для этого numpy. С numpy вы можете использовать numpy.square() который возвращает поэлементный квадрат ввода:
>>> import numpy as np
>>> start_list = [5, 3, 1, 2, 4]
>>> np.square(start_list)
array([25, 9, 1, 4, 16])
numpy.power()Существует также более общий numpy.power()
>>> np.power(start_list, 2)
array([25, 9, 1, 4, 16])
Просто используйте понимание списка:
start_list = [5, 3, 1, 2, 4]
squares = [x*x for x in start_list]
Примечание: в качестве небольшой оптимизации выполнение x*x выполняется быстрее, чем x**2 (или pow(x, 2)).
Вы можете использовать понимание списка:
[i**2 for i in start_list]
Или, если вы используете numpy, вы можете использовать tolist метод:
In [180]: (np.array(start_list)**2).tolist()
Out[180]: [25, 9, 1, 4, 16]
Или np.power:
In [181]: np.power(start_list, 2)
Out[181]: array([25, 9, 1, 4, 16], dtype=int32)
numpy, исходный вопрос импортируетnumpyи используетnumpy.array, также OP говорит, что исходит из фона MATLAB, кажется подходящим IMO. - person bakkal schedule 29.02.2016numpy. - person bakkal schedule 29.02.2016myarray ** 2- person seralouk schedule 27.11.2019