使用numpy创建ndarray
在本文中,我们将介绍如何使用numpy创建一个ndarray。
阅读更多:Numpy 教程
创建一个空的ndarray
我们可以使用numpy提供的empty
方法直接创建一个空的ndarray。该方法需要两个参数,第一个参数指定ndarray的形状,第二个参数指定数据类型。例如:
import numpy as np
arr = np.empty((3,4), int)
print(arr)
输出结果为:
[[0 0 0 0]
[0 0 0 0]
[0 0 0 0]]
使用列表创建ndarray
我们可以通过将一个列表传递给numpy的array
方法来创建一个ndarray。例如:
list = [1, 2, 3, 4, 5]
arr = np.array(list)
print(arr)
输出结果为:
[1 2 3 4 5]
我们还可以创建多维的ndarray。例如:
list = [[1, 2], [3, 4]]
arr = np.array(list)
print(arr)
输出结果为:
[[1 2]
[3 4]]
使用元组创建ndarray
我们可以通过将一个元组传递给numpy的array
方法来创建一个ndarray。例如:
tuple = (1, 2, 3, 4, 5)
arr = np.array(tuple)
print(arr)
输出结果为:
[1 2 3 4 5]
同样地,我们还可以创建多维的ndarray。例如:
tuple = ((1, 2), (3, 4))
arr = np.array(tuple)
print(arr)
输出结果为:
[[1 2]
[3 4]]
从字节数组创建ndarray
我们可以通过将一个字节数组传递给numpy的frombuffer
方法来创建一个ndarray。例如:
bytes = b'\x01\x02\x03\x04\x05'
arr = np.frombuffer(bytes, dtype=np.uint8)
print(arr)
输出结果为:
[1 2 3 4 5]
我们还可以指定字节数组的长度和数据类型。例如:
bytes = b'\x01\x02\x03\x04\x05\x06\x07\x08'
arr = np.frombuffer(bytes, dtype=np.uint16, count=4)
print(arr)
输出结果为:
[258 772 1286 1800]
总结
本文介绍了如何使用numpy创建一个ndarray。我们可以创建一个空的ndarray,使用列表或者元组创建一个ndarray,也可以从字节数组创建一个ndarray。numpy提供了多种创建ndarray的方法,可以根据我们的实际需求选择合适的方法。