创建一个for循环，用于从pd dataframe中提取特定行

Question

我有一个14列169882行的数据框。数据帧列头是

Columns: [Station, day, p1, p2, p3, p4, p5, p6, p7, p8, p9, p10, p11, Z]

在"Station“列中共有58个站点，它们的名称从1到58，但它们是随机分布在数据帧中的。

我写了下面的代码来从数据集中提取行和特定的站点名称(例如下面代码中的站点1)，并保存到具有对应站点名称的excel文件中：

grouped = df.groupby(df.Station)
S1= grouped.get_group(1)
S1.to_excel("1.xlsx") 

如何创建一个for循环来读取" station“列，提取与桩号1到58相关的行，然后将每个桩号输出保存到excel文件中？以一种更简单的方式，如何将上面的代码放在一个for循环中，以完成所有58个工作站的任务？

Answer 1

这可以通过两个步骤实现，获取站点的唯一列表，然后循环站点并在数据帧上应用过滤器

Station = df['Station'].unique().tolist()

for s in Station:
    file = s + '.csv'
    df1 = df[df.Station == str(s)]
    df1.to_csv(file, index=False)