我是初学者python用户,我在python2.7和python3.4.3上运行了以下代码
import matplotlib.pyplot as plt
import numpy as np
import scipy.stats as stats
alpha = 1
n = 100
u = stats.uniform(0,1)
F_inverse = lambda u: 1/alpha*np.log(1/(1-u))
v = np.array(map(F_inverse, u.rvs(n)))
print(v)
fig, ax = plt.subplots(1,1)
stats.probplot(v, (1,), dist='expon', plot=ax)
plt.show()
在python2上我得到一个很好的数组:
array([ 2.29133808e+00, 1.63236151e+00, 6.77776227e-01,
3.33668250e-01, 1.77830890e+00, 3.06193068e-01,
2.10677775e+00, 1.30525788e-01, 2.97056775e-01,
...
1.31463775e+00, 1.41840428e-03, 8.60594737e-01,
1.80644880e-01])
在python3上,我得到了这个:
array(<map object at 0x7f8aab6f3ef0>, dtype=object)
如果我改变了这个:
v = np.array(map(F_inverse, u.rvs(n)))
到
v = list(map(F_inverse, u.rvs(n)))
它在两者上工作正常,但我想要使用数组。 有没有办法让这个与np.array一起使用?
答案 0 :(得分:3)
将map object
转换为列表,然后将其传递给numpy.array
。
v = np.array(list(map(F_inverse, u.rvs(n))))
或者使用list comprehension而不是map来制作列表而不是map对象:
v = np.array([F_inverse(x) for x in u.rvs(n)])
但是,您不需要使用map
或列表理解不是必需的;只是直接调用F_inverse
就足够了,因为F_inverse
使用了矢量化操作:
v = F_inverse(u.rvs(n))