如何选择数据帧顶部行，其中一列的总和是原始列总和的90%？

import pandas as pd

# 假设数据帧名为df，列名为column_name
# 计算每列的总和
column_sums = df.sum()

# 排序总和列表
sorted_sums = column_sums.sort_values(ascending=False)

# 计算原始列总和的90%的值
target_sum = column_sums.sum() * 0.9

# 遍历排序后的列表，累加总和，找到满足条件的位置
cumulative_sum = 0
selected_index = 0
for index, value in sorted_sums.iteritems():
    cumulative_sum += value
    if cumulative_sum >= target_sum:
        selected_index = index
        break

# 打印所选行的位置
print("所选行的位置：", selected_index)