PYTHON如何使用numpy.where添加包含下一行数据帧中的数据的列？

import numpy as np

data = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])

next_row = np.roll(data, -1, axis=0)  # 将数据帧向上滚动一行
new_column = next_row[:, 0]  # 获取下一行数据帧中的第一列数据
result = np.where(data[:, 0] > 3, new_column, 0)  # 根据条件选择添加的数据

print(result)

import numpy as np

data = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])

next_row = np.roll(data, -1, axis=0)
new_column = next_row[:, 0]
result = np.where(data[:, 0] > 3, new_column, 0)

print(result)