numpy 报错 TypeError，如何解决？

Question

numpy 报错：TypeError: Field elements must be 2- or 3-tuples, got '5.0'

import numpy as np 
a = np.array([8.0,7.0,6.0],[5.0,4.0])
print(a)

我尝试运行此代码后，显示如下报错：

TypeError                                 Traceback (most recent call last)
<ipython-input-3-31ad187a1d2f> in <module>
      1 import numpy as np
----> 2 a = np.array([8.0,7.0,6.0],[5.0,4.0])
      3 print(a)

TypeError: Field elements must be 2- or 3-tuples, got '5.0'

求大佬解答！

Answer 1

https://numpy.org/doc/stable/...

np.array 的第二个参数是 dtype，描述数据的类型

你的调用 np.array([8.0, 7.0, 6.0], [5.0, 4.0]) 中，[5.0, 4.0] 是第二个参数

推测你想要写的是 np.array([[8.0, 7.0, 6.0], [5.0, 4.0]])，把两个列表放进一个大列表里，作为第一个参数