Python 将列表转换为二维 numpy 数组

Question

我有一些列表想转换成二维 numpy 数组。

list1 = [ 2, 7 , 8 , 5]
list2 = [18 ,29, 44,33]
list3 = [2.3, 4.6, 8.9, 7.7]

我想要的numpy数组是:

[[  2.   18.    2.3]
 [  7.   29.    4.6]
 [  8.   44.    8.9]
 [  5.   33.    7.7]]

我可以通过将列表中的各个项目直接输入到 numpy 数组表达式中作为 np.array(([2,18,2.3], [7,29, 4.6], [8,44 ,8.9], [5,33,7.7]), dtype=float).

但我希望能够将列表转换为所需的 numpy 数组。

Answer 1

一种方法是创建您的 numpy 数组，然后使用转置函数将其转换为您想要的输出:

import numpy as np

list1 = [ 2, 7 , 8 , 5]
list2 = [18 ,29, 44,33]
list3 = [2.3, 4.6, 8.9, 7.7]

arr = np.array([list1, list2, list3])
arr = arr.T
print(arr)

输出

[[  2.   18.    2.3]
 [  7.   29.    4.6]
 [  8.   44.    8.9]
 [  5.   33.    7.7]]