时间戳

首页 > 解决方案 > 用条件连接 csv 文件

python - 用条件连接 csv 文件

问题描述

我想以尽可能快的方式将多个 csv 文件与列值的条件连接起来。

我有一些有效的代码，但是在我将数据帧减少到我需要的站点（通过station_number列中的值）之前，它会连接所有 csv 文件的所有行。我想先选择我需要的行，然后再进行连接，这样可以提高运行时间。感谢您的任何建议！

station = int(input("station number? ")) 

def Datastations (station,path): 
    filepaths = [os.path.join(path, f) for f in listdir(path) if f.endswith('.csv')]
    df = pd.concat(map(pd.read_csv, filepaths)) 
    df = df[df.station_number==station]
    return (df)

df1 = Datastations(station,"refdata/obs") 
df2 = Datastations(station,"refdata/BoM_ETA_20160501-20170430/obs")

标签： pythonpandascsvconcatenation

解决方案

你没有说你遇到了什么麻烦，所以我只能为你重新排序：

import pandas as pd
import os

def Datastations (station,path): 
    filepaths = [os.path.join(path, f) for f in os.listdir(path) if f.endswith('.csv')]
    def process_csv(file_name):
        df = pd.read_csv(file_name)
        return df[df.station_number==station]
    return pd.concat(map(process_csv, filepaths))

推荐阅读