将带有条件 IF 的 groupby 元素/列表导出到 Excel 工作表

Question

我想检查一个“工作表”，如果它包含的内容超过例如如果有 250 个条目，我将创建一个新的 excel 表并将其保存在一个新文件中。

例如：

Leading-Zip:   Adresses that contains the Leading-Zip:
--------------------------

74                  400
73                  200   
72                   50

我使用这个命令来获取我想要分组的条目数：

worksheet['Zip-code-region'].value_counts()

我必须选择哪种方式来做到这一点？ 我必须创建一个列表吗？或者我可以使用带有 for 循环的命令吗？

尝试更新： 我正在导入一个 excel 文件：

xel = pd.read_excel(r'C:test.xlsx', sheet_name = None)

比我选择一张纸：

worksheet = xel[ws]

现在我添加一个新列 'leading-zip' 切片邮政编码：

worksheet['leading-zip']=worksheet['zip-code'].astype(str).str[:2].astype(int)

从那个'leading-zip'我想迭代每个'leading-zip' - 计算其中包含的地址，如果它们超过250，我想创建一个新的excel文件。

Answer 1

您可以过滤高于阈值的value_counts 结果，然后循环遍历它们的索引，将原始 DataFrame 中的相应子集保存为单独的 Excel 工作表：

import xlsxwriter
import numpy as np
import pandas as pd

df = pd.DataFrame({'zip': np.random.randint(10, 100, 1000)})

z = df['zip'].value_counts()

threshold = 15

writer = pd.ExcelWriter('output.xlsx', engine='xlsxwriter')
for i in z[z >= threshold].index:
    df[df['zip'] == i].to_excel(writer, str(i))

# save the remaining data as worksheet 'other':
df[df['zip'].isin(z[z < threshold].index)].to_excel(writer, 'other')

writer.save()