在特定列中保留元素最接近0的行

import numpy as np

# 假设数据集存储在一个二维数组中，命名为data
data = np.array([[1, 2, 3],
                 [4, 5, 6],
                 [7, 8, 9]])

column_index = 2  # 列索引，假设为2

min_diff = np.inf  # 初始化差值为正无穷
index_of_closest_row = -1  # 初始化最接近0的行的索引位置为-1

for i in range(len(data)):
    abs_diff = abs(data[i][column_index])  # 计算每行的绝对值
    if abs_diff < min_diff:
        min_diff = abs_diff
        index_of_closest_row = i

closest_row = data[index_of_closest_row]  # 最接近0的行

print("最接近0的行是:", closest_row)