将Python数组转换为NumPy数组


数组是一种数据结构,允许我们将相同数据类型元素存储在连续的内存块中。数组可以是一维、二维或三维,最多可达32维。

在Python中,创建数组的方法有很多种。一种方法是使用内置的`array`模块,它允许我们创建不同数据类型(如整数和浮点数)的数组。另一种方法是使用NumPy库,它提供了更强大和灵活的功能来实现数组。

使用`array`模块创建数组

Python中的内置模块`array`可以帮助我们创建不同维度的数组。

语法

以下是使用`array`模块的语法。

import array as arr
arr.array(datatype,list)

其中:

  • `array`是Python中的内置模块。

  • `arr`是`array`模块的别名。

  • `List`是所有元素的列表。

示例

在下面的示例中,我们通过将元素列表和所需的数据类型传递给`array`模块的`array()`函数来创建数组。此函数将返回一个包含指定元素的数组。

import array as arr
array_list = [22,24,14,12,7,2,1,5,21,11]
print("The list of elements to be used in array:",array_list)
created_arr = arr.array('i',array_list)
print("The created array using the specified list of elements:",created_arr)
print(type(created_arr))

输出

The list of elements to be used in array: [22, 24, 14, 12, 7, 2, 1, 5, 21, 11]
The created array using the specified list of elements: array('i', [22, 24, 14, 12, 7, 2, 1, 5, 21, 11])

示例

另一种创建多维数组的方法是使用`array`模块的`array`函数。

import array
data = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
arr = array.array('i', [elem for sublist in data for elem in sublist])
rows = len(data)
cols = len(data[0])
for i in range(rows):
    for j in range(cols):
        print(arr[i*cols + j], end=' ')
    print()

输出

1 2 3 
4 5 6 
7 8 9 

使用NumPy库创建数组

NumPy库提供了`array()`函数,可以帮助我们创建不同维度的数组。

语法

以下是使用NumPy库的语法。

numpy.array(list)

其中:

  • NumPy是一个库。

  • `array`是创建数组的函数。

  • `List`是元素列表。

示例

在下面的示例中,我们将使用NumPy库的`array()`函数创建一个一维数组,并将元素列表作为参数传递。

import numpy as np
list_elements = [232,34,23,98,48,43]
print("The list of elements to be used to create the array:",list_elements)
arr = np.array(list_elements)
print("The created array:",arr)
print("The dimension of the array:",np.ndim(arr))

输出

The list of elements to be used to create the array: [232, 34, 23, 98, 48, 43]
The created array: [232 34 23 98 48 43]
The dimension of the array: 1

示例

让我们再看一个例子,它通过将元素列表传递给NumPy库的`array()`函数来创建一个二维数组。

import numpy as np
list_elements = [[3,4],[2,3]]
print("The list of elements to be used to create the array:",list_elements)
arr = np.array(list_elements)
print("The created array:",arr)
print("The dimension of the array:",np.ndim(arr))

输出

The list of elements to be used to create the array: [[3, 4], [2, 3]]
The created array: [[3 4]
[2 3]]
The dimension of the array: 2

更新于:2023年8月9日

122 次浏览

启动你的职业生涯

完成课程获得认证

开始学习
广告