从numpy数组中选择特定的行和列

import numpy as np

# 创建一个示例数组
arr = np.array([[1, 2, 3],
                [4, 5, 6],
                [7, 8, 9]])

# 选择第1行和第2列
selected_element = arr[0, 1]
print("第1行第2列的元素:", selected_element)

# 选择第1行
selected_row = arr[0, :]
print("第1行:", selected_row)

# 选择第2列
selected_column = arr[:, 1]
print("第2列:", selected_column)

# 选择第1行到第2行，第1列到第2列
selected_subarray = arr[0:2, 0:2]
print("第1行到第2行，第1列到第2列的子数组:\n", selected_subarray)

# 错误示例
print(arr[3, 0])  # 索引超出范围

# 正确示例
if arr.shape[0] > 3:
    print(arr[3, 0])
else:
    print("索引超出范围")

# 错误示例
a = np.array([[1, 2], [3, 4]])
b = np.array([1, 2])
print(a + b)  # 形状不匹配

# 正确示例
b = np.array([[1], [2]])
print(a + b)  # 形状匹配