numpy-数组笔记

数组创建

import numpy as np

lst = [1, 2, 3]
print(np.array(lst))  # [1 2 3]
print(np.array([1.11, 2.22, 3.33]))  # [1.11 2.22 3.33]
print(np.array([[1.1, 1.2], [2.1, 2.2]]))  # 二维数组[[1.1 1.2] [2.1 2.2]]

数组的特殊创建函数

print(np.zeros(3))  # 一维数组[0. 0. 0.]
print(np.ones((2, 1), dtype=np.int_))  # 二维数组[[1][1]]
print(np.ones((2, 1), dtype=np.int_) * 15)  # 二维数组[[15][15]]
print(np.empty((2, 1)))  # [[ 8.18221536e+237][-1.85395876e+189]]

# 结构是2 行 2 列，内容是长度为3的数组
# 可以用于存放RGB图像，3表示3通道（BGR）
image = np.random.randint(0, 256, size=(2, 2, 3), dtype=np.uint8)  

数组运算

np1 = np.array([[1, 3]])
np2 = np.array([[3, 2]])

print(np1 + np2)  # [[4 5]]
print(np1 * np2)  # [[3 6]]
print(np1 / np2)  # [[0.33333333 1.5       ]]
print(np1 ** np2)  # 幂运算 [[1 9]]
print(np1 > np2)  # [[False  True]]