Разделить большой файл csv на основе даты в первом столбце Python 3.4.3

Question

Итак, я нашел часть ответа, который мне нужен здесь, в следующей ссылке, и он работает до тех пор, пока мои файлы csv находятся в формате 2015-03-01,1,2,3,1,3 для первого столбца. Как сохранить эту работу, когда первый столбец изменения в 2015-03-01 00:00:00.000

How to split a huge csv file based on content of first column?

import csv 
from itertools import groupby 

for key, rows in groupby(csv.reader(open("largeFile.csv", "r", encoding='utf-16')), 
        lambda row: row[0]): 
with open("%s.txt" % key, "w") as output: 
    for row in rows: 
     output.write(",".join(row) + "\n") 

Поэтому у меня есть один большой файл с примерно 1.7million строк в нем ...

2015.01.01,NULL,NULL,NULL,NULL,NULL,0,1,0,1,0,0,0,1 
2015.01.01,NULL,NULL,NULL,NULL,NULL,0,1,0,1,0,0,0,1 
2015.01.01,NULL,NULL,NULL,NULL,NULL,0,1,0,1,0,0,0,1 
2015.01.01,NULL,NULL,NULL,NULL,NULL,0,1,0,1,0,0,0,1 
2015.01.01,NULL,NULL,NULL,NULL,NULL,0,1,0,1,0,0,0,1 
2015.01.01,NULL,NULL,NULL,NULL,NULL,0,1,0,1,0,0,0,1 
2015.01.01,NULL,NULL,NULL,NULL,NULL,0,1,0,1,0,0,0,1 
2015.01.02,NULL,NULL,NULL,NULL,NULL,0,1,0,1,0,0,0,1 
2015.01.02,NULL,NULL,NULL,NULL,NULL,0,1,0,1,0,0,0,1 
2015.01.02,NULL,NULL,NULL,NULL,NULL,0,1,0,1,0,0,0,1 
2015.01.02,NULL,NULL,NULL,NULL,NULL,0,1,0,1,0,0,0,1 
2015.01.03,NULL,NULL,NULL,NULL,NULL,0,1,0,1,0,0,0,1 
2015.01.03,NULL,NULL,NULL,NULL,NULL,0,1,0,1,0,0,0,1 

И программа действительно создает новый текстовый документ для каждого дня, что здорово!

Но когда колонны выглядят следующим образом, он перестает работать.

2015-03-01 00:00:01.000,NULL,NULL,NULL,NULL,NULL,0,1,0,1,0,0,0,1 

2015-03-01 00:00:02.000,NULL,NULL,NULL,NULL,NULL,0,1,0,1,0,0,0,1 

2015-03-02 00:00:01.000,NULL,NULL,NULL,NULL,NULL,0,1,0,1,0,0,0,1 

2015-03-02 00:00:02.000,NULL,NULL,NULL,NULL,NULL,0,1,0,1,0,0,0,1 

2015-03-02 00:00:03.000,NULL,NULL,NULL,NULL,NULL,0,1,0,1,0,0,0,1 

2015-03-03 00:00:01.000,NULL,NULL,NULL,NULL,NULL,0,1,0,1,0,0,0,1 

2015-03-03 00:00:02.000,NULL,NULL,NULL,NULL,NULL,0,1,0,1,0,0,0,1 

, и это дает мне сообщение об ошибке.

Traceback (most recent call last): File "C:\Python34\Proj\documents\New folder\dataPullSplit2.py", line 6, in with open("%s.txt" % key, "w") as output: OSError: [Errno 22] Invalid argument: '2015-03-01 00:00:00.000.txt'

Может кто-нибудь указать мне в правильном направлении здесь, пожалуйста.

Found Temp Solution 

ОК, так что, изменив его с «ш» на «» Я теперь добавление к файлам и с помощью key[:-13] я смог отрезать штамп времени на имени файла ... это работает ... но это МЕДЛЕННО ... как я могу улучшить это и понять, почему он идет так медленно?

Вот код Сейчас

import csv 
from itertools import groupby 

for key, rows in groupby(csv.reader(open("asdf2.txt", "r", encoding='utf-16')), 
        lambda row: row[0]): 

with open("%s.txt" % key[:-13], "a") as output: 
    for row in rows: 
     output.write(",".join(row) + "\n") 

Answer 1

Предполагая, что ваши файлы должны сохранить шаблон 2015.01.01, чистка key должно работать:

key = key.split()[0].replace('-', '.') 

Полный код:

import csv 
from itertools import groupby 


def shorten_key(key): 
    return key.split()[0].replace('-', '.') 


for key, rows in groupby(csv.reader(open("asdf2.txt", "r", encoding='utf-16')), 
         lambda row: shorten_key(row[0])): 

    with open("%s.txt" % shorten_key(key), "a") as output: 
     for row in rows: 
      output.write(",".join(row) + "\n") 

Быстрый испытание:

keys = ['2015-03-01 00:00:02.000', '2015.01.01'] 

for key in keys: 
    print(key.split()[0].replace('-', '.')) 

Выход:

2015.03.01 
2015.01.01