将NumPy数组转换为列表|极客教程

将NumPy数组转换为列表

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

1. 将一维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数组转换为列表

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

接下来，我们看看如何将二维NumPy数组转换为列表。同样，我们可以使用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数组转换为列表

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

除了一维和二维数组，我们还可以将多维NumPy数组转换为列表。同样，我们可以使用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数组转换为列表

4. 将NumPy数组中的字符串转换为列表

有时候，NumPy数组中的元素是字符串，我们也可以将这些字符串转换为列表。同样，我们可以使用tolist()方法来实现这个转换。

import numpy as np

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

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

print(arr_list)

Output:

将NumPy数组转换为列表

5. 将NumPy数组中的布尔值转换为列表

除了字符串，NumPy数组中的元素还可以是布尔值。我们也可以将这些布尔值转换为列表。

import numpy as np

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

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

print(arr_list)

Output:

将NumPy数组转换为列表

6. 将NumPy数组中的复数转换为列表

NumPy数组中的元素还可以是复数。我们也可以将这些复数转换为列表。

import numpy as np

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

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

print(arr_list)

Output:

将NumPy数组转换为列表

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

有时候，NumPy数组中的元素可能是NaN值（Not a Number）。我们也可以将这些NaN值转换为列表。

import numpy as np

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

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

print(arr_list)

Output:

将NumPy数组转换为列表

8. 将NumPy数组中的无穷大值转换为列表

除了NaN值，NumPy数组中的元素还可能是无穷大值。我们也可以将这些无穷大值转换为列表。

import numpy as np

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

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

print(arr_list)

Output:

将NumPy数组转换为列表

9. 将NumPy数组中的日期时间转换为列表

在处理时间序列数据时，NumPy数组中的元素可能是日期时间。我们也可以将这些日期时间转换为列表。

import numpy as np

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

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

print(arr_list)

Output:

将NumPy数组转换为列表

10. 将NumPy数组中的自定义对象转换为列表

除了基本数据类型，NumPy数组中的元素还可以是自定义对象。我们也可以将这些自定义对象转换为列表。

import numpy as np

# 定义一个自定义对象
class Person:
    def __init__(self, name, age):
        self.name = name
        self.age = age

# 创建一个包含自定义对象的NumPy数组
arr = np.array([Person('Alice', 30), Person('Bob', 25)])

# 将NumPy数组中的自定义对象转换为列表
arr_list = [obj.__dict__ for obj in arr]

print(arr_list)