python数组和列表的转换

Python中，列表和数组可以相互转换。

1. 列表转数组

使用NumPy库中的array()函数，可以将列表转换为数组。

示例代码：

import numpy as np

list1 = [1, 2, 3, 4, 5]
arr1 = np.array(list1)

print(arr1)

输出结果：

[1 2 3 4 5]

2. 数组转列表

使用tolist()方法，可以将数组转换为列表。

示例代码：

import numpy as np

arr2 = np.array([6, 7, 8, 9, 10])
list2 = arr2.tolist()

print(list2)

输出结果：

[6, 7, 8, 9, 10]

注意：如果数组中有多维数据，转换为列表时需要注意维度的变化。例如，一个二维数组转换为列表时，每行数据会作为一个子列表。

示例代码：

import numpy as np

arr3 = np.array([[1, 2, 3], [4, 5, 6]])
list3 = arr3.tolist()

print(list3)

输出结果：

[[1, 2, 3], [4, 5, 6]]