[已解决]载入numpy后的数据类型问题

Pliosauroidea

在学习matplotlib时候我试图调用np.random.normal生成一组随机点阵用于绘图，在调用时传入了int类型的参数x，但是当我将x*2传入normal函数时报错，具体情况如下：

1. def test_random(number):
   
2.     M = np.random.normal(0, 1, 2 * number)
   
3.     M.reshape((2, number))
   
4.     print(M[0])
   
5.     return M

复制代码

对应的调用为：

1. M = test_random(int(input('scale:\n')))

复制代码

此时报错：

1. Traceback (most recent call last):
   
2.   File "F:/pytest/matplotlib_1.py", line 61, in <module>
   
3.     main()
   
4.   File "F:/pytest/matplotlib_1.py", line 53, in main
   
5.     Z = test_random(M)
   
6.   File "F:/pytest/matplotlib_1.py", line 41, in test_random
   
7.     M = np.random.normal(0, 1, 2 * number)
   
8.   File "mtrand.pyx", line 1497, in numpy.random.mtrand.RandomState.normal
   
9.   File "_common.pyx", line 577, in numpy.random._common.cont
   
10. TypeError: 'numpy.float64' object cannot be interpreted as an integer

复制代码

我又试图将2*number强制转化为int：

1. def test_random(number):
   
2.     double_number=int(2*number)
   
3.     M = np.random.normal(0, 1, double_number)
   
4.     M.reshape((2, number))
   
5.     print(M[0])
   
6.     return M

复制代码

此时报错：

1.   File "F:/pytest/matplotlib_1.py", line 54, in main
   
2.     Z = test_random(M)
   
3.   File "F:/pytest/matplotlib_1.py", line 41, in test_random
   
4.     double_number=int(2*number)
   
5. TypeError: only size-1 arrays can be converted to Python scalars

复制代码

请问要如何才能将2*number正确的传入normal函数中呢

最佳答案

月排行榜 / 总排行榜

疾风怪盗

2020-9-18 16:47:53

看不懂你的意思，按你的代码，输入14，运行成功了啊，你要输入什么数？

1. import numpy as np
   
2. 
   
3. def test_random(number):
   
4.     M = np.random.normal(0, 1, 2 * number)
   
5.     M.reshape((2, number))
   
6.     print(M[0])
   
7.     return M
   
8. 
   
9. 
   
10. M = test_random(int(input('scale:\n')))
    
11. print(M)

复制代码

scale:
14
-0.009342555457168914
[-0.00934256 -0.56134032  0.93871941  0.7672819  -0.06824285 -0.05190735
  1.74827189  0.7951845   0.40550305  0.46697624 -0.59951913  1.39413893
  0.82887424 -1.87081902  0.82180017  0.46254866 -0.21679419 -0.14561513
  0.74897389  0.32006024  0.82306899  1.02708188  0.31247572  0.17292185
  2.01692653  0.7196225  -0.36370474 -0.21091174]

Process finished with exit code 0

跳转到最佳答案楼层

疾风怪盗

看不懂你的意思，按你的代码，输入14，运行成功了啊，你要输入什么数？

1. import numpy as np
   
2. 
   
3. def test_random(number):
   
4.     M = np.random.normal(0, 1, 2 * number)
   
5.     M.reshape((2, number))
   
6.     print(M[0])
   
7.     return M
   
8. 
   
9. 
   
10. M = test_random(int(input('scale:\n')))
    
11. print(M)

复制代码

scale:
14
-0.009342555457168914
[-0.00934256 -0.56134032  0.93871941  0.7672819  -0.06824285 -0.05190735
  1.74827189  0.7951845   0.40550305  0.46697624 -0.59951913  1.39413893
  0.82887424 -1.87081902  0.82180017  0.46254866 -0.21679419 -0.14561513
  0.74897389  0.32006024  0.82306899  1.02708188  0.31247572  0.17292185
  2.01692653  0.7196225  -0.36370474 -0.21091174]

Process finished with exit code 0

Pliosauroidea

疾风怪盗 发表于 2020-9-18 16:47
看不懂你的意思，按你的代码，输入14，运行成功了啊，你要输入什么数？

迷惑。。我自己用python3.7.0跑会报错emm等下我把整块都贴上来看一下orz

Pliosauroidea

Pliosauroidea 发表于 2020-9-18 17:10
迷惑。。我自己用python3.7.0跑会报错emm等下我把整块都贴上来看一下orz

啊绝了，原来是我调用时候不小心把返回值重新调用了一次，导致报错emmmm