我有一个2D numpy数组:
>>> arr = np.arange(1,10).reshape((3,3))
array([[0, 1, 2],
[3, 4, 5],
[6, 7, 8]])
我希望获取数组中所有元素的对数。 以下代码有效,但有点笨重
>>> from math import log10
>>> logArr = np.empty((3,3))
>>> for i in range(3):
... for j in range(3):
... logArr[i][j] = log10(arr[i][j])
...
array([[ 0. , 0.30103 , 0.47712125],
[ 0.60205999, 0.69897 , 0.77815125],
[ 0.84509804, 0.90308999, 0.95424251]])
是否存在更有效/'pythonic'的方式进行此类操作?
答案 0 :(得分:2)
有一个numpy函数,请尝试numpy.log
>>> arr = np.arange(1,10).reshape((3,3))
>>> np.log(arr)
array([[ 0. , 0.69314718, 1.09861229],
[ 1.38629436, 1.60943791, 1.79175947],
[ 1.94591015, 2.07944154, 2.19722458]])
或者在您的实施中,您可以使用numpyp.log10查找基数为10的日志。
>>> np.log10(arr)
array([[ 0. , 0.30103 , 0.47712125],
[ 0.60205999, 0.69897 , 0.77815125],
[ 0.84509804, 0.90308999, 0.95424251]])