从现有的csv文件python pandas创建多个csv文件

时间:2018-10-20 00:54:44

标签: python pandas csv pandas-groupby

我正在尝试获取一个较大的csv文件,并为两列排序编写一个csv文件。我能够从文件中获得两个单独的唯一值,从而能够知道需要创建哪些csv文件。
防爆数据:

1,224939.203,1243008.651,1326.774,F,C-GRAD-FILL,09/22/18 07:24:34,
1,225994.242,1243021.426,1301.772,BS,C-GRAD-FILL,09/24/18 08:24:18,
451,225530.332,1243016.186,1316.173,GRD,C-TOE,10/02/18 11:49:13,
452,225522.429,1242996.017,1319.168,GRD,C-TOE KEY,10/02/18 11:49:46,

我想创建一个csv文件“ C-GRAD-FILL 09-22-18.csv”,其中包含与这两个值匹配的所有数据。我无法决定如何遍历两个值的数据。

def readData(fileName):
    df = pd.read_csv(fileName,index_col=False, names+['Number','Northing','Easting','Elevation','Description','Layer','Date'],parse_dates=['Date'] )
    ##Layers here!!!
    layers = df['Layer'].unique()
    ##Dates here!!! AS DATETIME OBJECTS!!!!
    dates = df['Date'].map(lambda t: t.date()).unique()
    ##Sorted in order
    sortedList = df.sort_values(by=['Layer','Date'])

1 个答案:

答案 0 :(得分:1)

您可以使用GroupBy对象。首先,请确保您的日期采用正确的字符串格式:

df['Date'] = pd.to_datetime(df['Date']).dt.strftime('%m-%d-%y')

要输出所有文件,请迭代GroupBy对象:

for (layer, date), group in df.groupby(['Layer', 'Date']):
    group.to_csv(f'{layer} {date}.csv', index=False)

或者,对于一种特定的组合:

layer = 'C-GRAD-FILL'
date = '09-22-18'
g = df.groupby(['Layer', 'Date'])

g.get_group((layer, date)).to_csv(f'{layer} {date}.csv', index=False)
相关问题