将列表转换为 Numpy ndarray

Question

我想转换这个列表

li = [3, 2 , 1 , 4]

到以下 NumPy ndarray

[[3], [2], [1], [4]]

我尝试过，np.asarray()但它没有将其转换为ndarray. 有没有办法提到轴？

Answer 1

您可以使用列表推导式创建嵌套列表，然后使用numpy.array().

>>> import numpy as np
>>> li = [3, 2, 1, 4]

>>> np.array([[x] for x in li])    
[[3]
 [2]
 [1]
 [4]]