NumPy转为list|极客教程

NumPy转为list

在数据处理和科学计算中，NumPy是一个非常强大的库，它提供了高性能的多维数组对象和各种工具，可以用来处理大型数据集和执行各种数学运算。有时候我们需要将NumPy数组转换为Python中的列表（list），以便更方便地进行操作或者与其他Python库进行交互。本文将详细介绍如何将NumPy数组转换为列表，并提供多个示例代码。

1. 将一维NumPy数组转为列表

首先，我们来看如何将一维的NumPy数组转换为Python中的列表。可以使用tolist()方法来实现这一转换。

import numpy as np

# 创建一个一维NumPy数组
arr = np.array([1, 2, 3, 4, 5])

# 将NumPy数组转换为列表
arr_list = arr.tolist()

print(arr_list)

Output:

NumPy转为list

2. 将二维NumPy数组转为列表

接下来，我们看看如何将二维的NumPy数组转换为Python中的列表。同样可以使用tolist()方法来实现。

import numpy as np

# 创建一个二维NumPy数组
arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])

# 将NumPy数组转换为列表
arr_list = arr.tolist()

print(arr_list)

Output:

NumPy转为list

3. 将多维NumPy数组转为列表

除了一维和二维数组，我们还可以将多维的NumPy数组转换为Python中的列表。同样使用tolist()方法即可。

import numpy as np

# 创建一个三维NumPy数组
arr = np.array([[[1, 2], [3, 4]], [[5, 6], [7, 8]]])

# 将NumPy数组转换为列表
arr_list = arr.tolist()

print(arr_list)

Output:

NumPy转为list

4. 将NumPy数组中的元素转为列表

有时候我们不需要将整个NumPy数组转为列表，而是只需要将数组中的元素转为列表。可以使用tolist()方法来实现。

import numpy as np

# 创建一个一维NumPy数组
arr = np.array([1, 2, 3, 4, 5])

# 将NumPy数组中的元素转换为列表
arr_list = arr.tolist()

print(arr_list)

Output:

NumPy转为list

5. 将NumPy数组中的字符串元素转为列表

如果NumPy数组中包含字符串元素，同样可以使用tolist()方法将其转换为列表。

import numpy as np

# 创建一个包含字符串的NumPy数组
arr = np.array(['geek-docs.com', 'numpy', 'list'])

# 将NumPy数组中的字符串元素转换为列表
arr_list = arr.tolist()

print(arr_list)

Output:

NumPy转为list

6. 将NumPy数组中的布尔值元素转为列表

如果NumPy数组中包含布尔值元素，同样可以使用tolist()方法将其转换为列表。

import numpy as np

# 创建一个包含布尔值的NumPy数组
arr = np.array([True, False, True])

# 将NumPy数组中的布尔值元素转换为列表
arr_list = arr.tolist()

print(arr_list)

Output:

NumPy转为list

7. 将NumPy数组中的复数元素转为列表

如果NumPy数组中包含复数元素，同样可以使用tolist()方法将其转换为列表。

import numpy as np

# 创建一个包含复数的NumPy数组
arr = np.array([1+2j, 3+4j, 5+6j])

# 将NumPy数组中的复数元素转换为列表
arr_list = arr.tolist()

print(arr_list)

Output:

NumPy转为list

8. 将NumPy数组中的NaN值转为列表

如果NumPy数组中包含NaN值（Not a Number），同样可以使用tolist()方法将其转换为列表。

import numpy as np

# 创建一个包含NaN值的NumPy数组
arr = np.array([1, np.nan, 3])

# 将NumPy数组中的NaN值转换为列表
arr_list = arr.tolist()

print(arr_list)

Output:

NumPy转为list

9. 将NumPy数组中的inf值转为列表

如果NumPy数组中包含inf值（无穷大），同样可以使用tolist()方法将其转换为列表。

import numpy as np

# 创建一个包含inf值的NumPy数组
arr = np.array([1, np.inf, 3])

# 将NumPy数组中的inf值转换为列表
arr_list = arr.tolist()

print(arr_list)

Output:

NumPy转为list

10. 将NumPy数组中的日期时间元素转为列表

如果NumPy数组中包含日期时间元素，同样可以使用tolist()方法将其转换为列表。

import numpy as np

# 创建一个包含日期时间元素的NumPy数组
arr = np.array(['2022-01-01', '2022-02-01', '2022-03-01'], dtype='datetime64')

# 将NumPy数组中的日期时间元素转换为列表
arr_list = arr.tolist()

print(arr_list)