如何依据DataFrame首列最小值拆分数据集？

错误原因 #

你当前代码中pd.DataFrame.idxmin(data)返回的是一个Series对象（存储了DataFrame每一列最小值对应的索引），而iloc进行位置切片时需要的是单个整数索引，因此触发了TypeError。

正确实现步骤 #

1. 精准定位首列最小值的位置 #

只针对目标列（x列）调用idxmin()，这样会直接返回单个整数索引：

min_idx = data['x'].idxmin()

如果x列存在多个相同的最小值，idxmin()仅返回第一个出现的位置。若需获取所有最小值的索引，可改用：

min_indices = data[data['x'] == data['x'].min()].index

2. 拆分数据集 #

使用获取到的整数索引进行切片：

• 从起始行到最小值位置（包含该行）：

lower_surface = data.iloc[:min_idx + 1]

• 从最小值位置到末尾行（包含该行）：

upper_surface = data.iloc[min_idx:]

完整修正代码 #

import pandas as pd

# 假设你的原始数据已加载为data
data = pd.DataFrame(data)

# 获取x列最小值的位置
min_idx = data['x'].idxmin()

# 拆分数据集
lower_surface = data.iloc[:min_idx + 1]
upper_surface = data.iloc[min_idx:]

补充说明 #

你的示例数据中x列从1下降到最小值后回升，拆分后lower_surface会包含从起始到最小值的下降段数据，upper_surface包含从最小值到末尾的回升段数据。

内容的提问来源于stack exchange，提问作者John278