Python——如何獲取二維列表的列數據

                                                    Python——如何獲取二維列表的列數據

目標：如何簡單快速獲取二維列表的列數據。

代碼：

import numpy as np

test_data = [[1, 2, 3], [4, 5, 6]]
print("二維數據:", test_data)
print("第一列數據:", list(np.array(test_data).T[0]))
print("第二列數據:", list(np.array(test_data).T[1]))
print("第三列數據:", list(np.array(test_data).T[2]))

運行結果：

二維數據: [[1, 2, 3], [4, 5, 6]]
第一列數據: [1, 4]
第二列數據: [2, 5]
第三列數據: [3, 6]