Numpy操作

import numpy as np

# 从Python列表创建数组
arr = np.array([1, 2, 3, 4, 5])

# 创建全0数组
zeros_arr = np.zeros((3, 4))

# 创建全1数组
ones_arr = np.ones((2, 3))

# 创建等差数列
arange_arr = np.arange(0, 10, 2)

a = np.array([1, 2, 3])
b = np.array([4, 5, 6])

# 加法
sum_arr = a + b

# 乘法
product_arr = a * b

# 点积
dot_product = np.dot(a, b)

# 创建包含NaN的数组
arr_with_nan = np.array([1, 2, np.nan, 4])

# 检查并去除NaN值
clean_arr = arr_with_nan[~np.isnan(arr_with_nan)]

# 创建一个一维数组
original_arr = np.arange(12)

# 重塑为3x4的二维数组
reshaped_arr = original_arr.reshape((3, 4))

a = np.array([[1, 2], [3, 4]])
b = np.array([[5, 6]])

# 垂直拼接
vertically_stacked = np.vstack((a, b))

# 水平拼接
horizontally_stacked = np.hstack((a, b.T))