Python:读取我的文件夹中的多个但不是全部的csv文件

时间:2018-08-24 17:49:08

标签: python csv

我想从我的文件夹中读取一些csv文件,并将它们连接到一个大熊猫数据框。我所有的csv文件都以数字结尾,而我只想读取数字以(6〜10,16〜20,26〜30)结尾的文件。我的目标是迭代读取文件。到目前为止,是我的代码:

import pandas as pd

data_one = pd.read_csv('Datafile-6.csv', header=None)
for i in range(7,11):
    data99 = pd.read_csv('Datafile-'+i+'*.csv', header=None) #this line needs work
    data_one = pd.concat([data_one, data99.iloc[:,1]],axis=1,ignore_index=True)

data_two = pd.read_csv('Datafile-16.csv', header=None)
for j in range(17,21):
    #Repeat similar process

我应该如何处理“ data99”,以使“ data_one”包含从“ Datafile-6”到“ Datafile-10”的列?

从数据文件6-10中获取数据后,data_one的前五行应如下所示。

      0         1         2         3         4         5
0 -40.0  0.179836  0.179630  0.179397  0.179192  0.179031
1 -39.0  0.183696  0.183441  0.183204  0.182977  0.182795
2 -38.0  0.186720  0.186446  0.186191  0.185949  0.185762
3 -37.0  0.189490  0.189207  0.188935  0.188686  0.188475
4 -36.0  0.192154  0.191851  0.191569  0.191301  0.191086

第0列包含在所有数据文件中,因此我仅将所有后续数据文件的第1列串联起来。

1 个答案:

答案 0 :(得分:0)

您需要使用glob模块:

import glob, os
import pandas as pd

path =r'C:\YourFolder' #path to folder with .csv files
all = glob.glob(path + "/*.csv")
d_frame = pd.DataFrame()
list_ = []
for file_ in all:
    df = pd.read_csv(file_,index_col=None, header=0)
    if df['YourColumns'].tail(1).isin([6,7,8,9,10,16,17,18,19,20,26,27,28,29,30]) == True: #You can modify list with conditions you need
        list_.append(df)
d_frame = pd.concat(list_)