当前位置:首页python > 正文

用python读取一个文件夹下的所有CSV文件里某一列数据中最大值,将此最大值所在行截取到新CSV文件中?

作者:野牛程序员:2023-11-21 12:18:14python阅读 2953

用python读取一个文件夹下的所有CSV文件里某一列数据中最大值,将此最大值所在行截取到新CSV文件中?

import os
import pandas as pd

def find_max_value_row(csv_file):
    # 读取CSV文件
    df = pd.read_csv(csv_file)

    # 找到某一列数据中的最大值
    max_value = df['your_column_name'].max()

    # 找到最大值所在行的索引
    max_value_index = df.index[df['your_column_name'] == max_value].tolist()[0]

    # 截取最大值所在行,并保存到新CSV文件
    new_df = df.loc[[max_value_index]]
    new_csv_file = 'new_file.csv'
    new_df.to_csv(new_csv_file, index=False)

    return new_csv_file

# 遍历文件夹下所有CSV文件
folder_path = 'your_folder_path'
for filename in os.listdir(folder_path):
    if filename.endswith('.csv'):
        csv_file_path = os.path.join(folder_path, filename)
        new_csv_file_path = find_max_value_row(csv_file_path)
        print(f"处理文件 {filename} 完成,生成新文件 {new_csv_file_path}")


请替换代码中的 'your_column_name' 为想要查找最大值的列的名称,'your_folder_path' 为包含CSV文件的文件夹路径。这段代码将遍历文件夹下的所有CSV文件,找到每个文件中指定列的最大值所在行,并将该行保存到新的CSV文件中。

野牛程序员教少儿编程与信息学奥赛-微信|电话:15892516892
野牛程序员教少儿编程与信息学竞赛-微信|电话:15892516892
相关推荐

最新推荐

热门点击