Как удалить определенную строку, общую для нескольких строк в файле CSV, с помощью скрипта Python?

У меня есть файл csv, который содержит 65000 строк (размер примерно 28 МБ). В каждой из строк дан определенный путь в начале, т.е. "c:\abc\bcd\def\123\456". Теперь допустим, что путь "c:\abc\bcd\" общий во всех строках, а остальное содержимое разное. Мне нужно удалить общую часть (в данном случае "c:\abc\bcd\") из всех строк с помощью скрипта Python. Например, содержимое CSV-файла соответствует упомянутому.

C:/Abc/bcd/Def/Test/temp/test/GLNext/FILE0.frag                   0   0   0
C:/Abc/bcd/Def/Test/temp/test/GLNext/FILE0.vert                   0   0   0
C:/Abc/bcd/Def/Test/temp/test/GLNext/FILE0.link-link-0.frag       16  24  3
C:/Abc/bcd/Def/Test/temp/test/GLNext/FILE0.link-link-0.vert       87  116 69
C:/Abc/bcd/Def/Test/temp/test/GLNext/FILE0.link-link-0.vert.bin   75  95  61
C:/Abc/bcd/Def/Test/temp/test/GLNext/FILE0.link-link-0            0   0
C:/Abc/bcd/Def/Test/temp/test/GLNext/FILE0.link-link-6            0   0   0 

В приведенном выше примере мне нужен вывод, как показано ниже

FILE0.frag                  0   0   0
FILE0.vert                  0   0   0
FILE0.link-link-0.frag      17  25  2
FILE0.link-link-0.vert      85  111 68
FILE0.link-link-0.vert.bin  77  97  60
FILE0.link-link-0               0   0
FILE0.link                  0   0   0

Может ли кто-нибудь из вас помочь мне с этим?


person Anurag Tiwary    schedule 06.07.2015    source источник
comment
что вы пробовали до сих пор?   -  person ozgur    schedule 06.07.2015
comment
Как насчет удаления всех символов до последней обратной косой черты?   -  person Avinash Raj    schedule 06.07.2015
comment
из вопроса не ясно, должна ли общая часть определяться автоматически?   -  person Henrik    schedule 06.07.2015
comment
Привет Авинаш. Спасибо за ответ. Я новичок в python и пытаюсь понять, как python работает с файлами. Поскольку эта операция должна выполняться с несколькими файлами в каталоге и подкаталогах, я использую для этого os.walk.   -  person Anurag Tiwary    schedule 06.07.2015
comment
Хенрик, как насчет того, чтобы сохранить общую часть в какой-то переменной, потому что она все время будет одной и той же?   -  person Anurag Tiwary    schedule 06.07.2015
comment
Если какой-либо из ответов помог вам, рекомендуется отметить этот ответ как принятый, чтобы он мог помочь другим с вопросом, похожим на ваш.   -  person Alok Mysore    schedule 06.07.2015
comment
Привет, Алок, я все еще работаю над этим и пробую разные ответы. Отмечу ответ как принятый, как только закончу. Большое спасибо за Вашу помощь.   -  person Anurag Tiwary    schedule 06.07.2015


Ответы (6)


^\S+/

Вы можете просто использовать это регулярное выражение для каждой строки и заменить на empty string. См. демо.

https://regex101.com/r/cK4iV0/17

import re
p = re.compile(ur'^\S+/', re.MULTILINE)
test_str = u"C:/Abc/Def/Test/temp/test/GLNext/FILE0.frag                   0   0   0\nC:/Abc/Def/Test/temp/test/GLNext/FILE0.vert                   0   0   0\nC:/Abc/Def/Test/temp/test/GLNext/FILE0.link-link-0.frag       16  24  3\nC:/Abc/Def/Test/temp/test/GLNext/FILE0.link-link-0.vert       87  116 69\nC:/Abc/Def/Test/temp/test/GLNext/FILE0.link-link-0.vert.bin   75  95  61\nC:/Abc/Def/Test/temp/test/GLNext/FILE0.link-link-0            0   0\nC:/Abc/Def/Test/temp/test/GLNext/FILE0.link-link-6            0   0   0 "
subst = u" "

result = re.sub(p, subst, test_str)
person vks    schedule 06.07.2015

Как насчет чего-то вроде,

import csv

with open("file.csv", 'rb') as f:
    sl = []
    csvread = csv.reader(f, delimiter=' ')
    for line in csvread:
        sl.append(line.replace("C:/Abc/Def/Test/temp\.\test\GLNext\", ""))

Чтобы записать список sl для filenew использования,

with open('filenew.csv', 'wb') as f:
    csvwrite = csv.writer(f, delimiter=' ')
    for line in sl:
        csvwrite.writerow(line)
person Ed Smith    schedule 06.07.2015
comment
Спасибо за ответ Эд. Это также сохранит файл? - person Anurag Tiwary; 06.07.2015
comment
У вас будет список новых строк в sl. надо позвонить написать. Если ваш файл CSV (с разделителем, а не данными, которые у вас есть выше), вам следует использовать csv.reader и csv.writer. - person Ed Smith; 06.07.2015

Вы можете автоматически определять общий префикс без необходимости его жесткого кодирования. Вам действительно не нужно regex для этого. Вместо этого можно использовать os.path.commonprefix:

import csv
import os

with open('data.csv', 'rb') as csvfile:
    reader = csv.reader(csvfile)
    paths = [] #stores all paths
    rows = [] #stores all lines
    for row in reader:
        paths.append(row[0].split("/")) #split path by "/"
        rows.append(row)

    commonprefix = os.path.commonprefix(paths) #finds prefix common to all paths

    for row in rows:
        row[0] = row[0].replace('/'.join(commonprefix)+'/', "") #remove prefix

rows теперь есть список списков, которые вы можете записать в файл

with open('data2.csv', 'wb') as csvfile:
    writer = csv.writer(csvfile)
    for row in rows:
        writer.writerow(row)
person Alok Mysore    schedule 06.07.2015

Следующий скрипт Python прочитает ваш файл (при условии, что он выглядит как ваш пример) и создаст версию, удалив общие папки:

import os.path, csv

finput = open("d:\\input.csv","r")
csv_input = csv.reader(finput, delimiter=" ", skipinitialspace=True)
csv_output = csv.writer(open("d:\\output.csv", "wb"), delimiter=" ")

# Create a set of unique folder names

set_folders = set()

for input_row in csv_input:
    set_folders.add(os.path.split(input_row[0])[0])

# Determine the common prefix

base_folder = os.path.split(os.path.commonprefix(set_folders))[0]
nprefix = len(base_folder) + 1

# Go back to the start of the input CSV 

finput.seek(0)

for input_row in csv_input:
    csv_output.writerow([input_row[0][nprefix:]] + input_row[1:])

Используя в качестве входных данных следующее:

C:/Abc/Def/Test/temp/test/GLNext/FILE0.frag                   0   0   0
C:/Abc/Def/Test/temp/test/GLNext/FILE0.vert                   0   0   0
C:/Abc/Def/Test/temp/test/GLNext/FILE0.link-link-0.frag       16  24  3
C:/Abc/Def/Test/temp/test/GLNext2/FILE0.link-link-0.vert       87  116 69
C:/Abc/Def/Test/temp/test/GLNext5/FILE0.link-link-0.vert.bin   75  95  61
C:/Abc/Def/Test/temp/test/GLNext7/FILE0.link-link-0            0   0
C:/Abc/Def/Test/temp/test/GLNext/FILE0.link-link-6            0   0   0

Результат выглядит следующим образом:

GLNext/FILE0.frag 0 0 0
GLNext/FILE0.vert 0 0 0
GLNext/FILE0.link-link-0.frag 16 24 3
GLNext2/FILE0.link-link-0.vert 87 116 69
GLNext5/FILE0.link-link-0.vert.bin 75 95 61
GLNext7/FILE0.link-link-0 0 0
GLNext/FILE0.link-link-6 0 0 0

С одним пробелом между каждым столбцом, хотя это можно легко изменить.

person Martin Evans    schedule 06.07.2015

Итак, я попробовал что-то вроде этого

for dirName, subdirList, fileList in os.walk(Directory):
    for fname in fileList:
        if fname.endswith('.csv'):
            for line in fileinput.input(os.path.join(dirName, fname), inplace = 1):
                location = line.find(r'GLNext')
                if location > 0:
                    location += len('GLNext')
                    print line.replace(line[:location], ".")
                else:
                    print line
person Anurag Tiwary    schedule 08.07.2015

Для этого вы можете использовать библиотеку pandas. При этом вы можете использовать pandas потрясающую обработку больших CSV-файлов (даже сотнями МБ).

Код:

import pandas as pd

csv_file = 'test_csv.csv'
df = pd.read_csv(csv_file, header=None)
print df
print "-------------------------------------------"

path = "C:/Abc/bcd/Def/Test/temp/test/GLNext/"
df[0] = df[0].replace({path:""}, regex=True)

print df
# df.to_csv("truncated.csv") # Export to new file.

Результат:

                                                   0   1    2   3
0    C:/Abc/bcd/Def/Test/temp/test/GLNext/FILE0.frag   0    0   0
1    C:/Abc/bcd/Def/Test/temp/test/GLNext/FILE0.vert   0    0   0
2  C:/Abc/bcd/Def/Test/temp/test/GLNext/FILE0.lin...  16   24   3
3  C:/Abc/bcd/Def/Test/temp/test/GLNext/FILE0.lin...  87  116  69
4  C:/Abc/bcd/Def/Test/temp/test/GLNext/FILE0.lin...  75   95  61
5  C:/Abc/bcd/Def/Test/temp/test/GLNext/FILE0.lin...   0    0 NaN
6  C:/Abc/bcd/Def/Test/temp/test/GLNext/FILE0.lin...   0    0   0
-------------------------------------------
                            0   1    2   3
0                  FILE0.frag   0    0   0
1                  FILE0.vert   0    0   0
2      FILE0.link-link-0.frag  16   24   3
3      FILE0.link-link-0.vert  87  116  69
4  FILE0.link-link-0.vert.bin  75   95  61
5           FILE0.link-link-0   0    0 NaN
6           FILE0.link-link-6   0    0   0
person NullDev    schedule 08.07.2015