Python 2를 사용하여 CSV 파일 읽기

Question

저는 Python 2.7을 사용하고 있습니다. 나는 Python을 처음 접했을 뿐이다. CSV 파일 (값은 공백으로 구분됨)을 읽고 좌표 위의 머리글을 기준으로 값을 분리하려고합니다. 파일 형식은 필자가 익숙한 것이 아니며 값을 올바르게 읽는 데 문제가 있습니다. 비록 내가 올바르게 읽을 수 있다고해도, 나는 그것을 어떻게 목록에 넣는 지 이해하지 못합니다.

class CommentedFile: 
    def __init__(self, f, commentstring="#"): 
     self.f = f 
     self.commentstring = commentstring 
    def next(self): 
     line = self.f.next() 
     while line.startswith(self.commentstring): 
      line = self.f.next() 
     return line 
    def __iter__(self): 
     return self 

#I did this in order to ignore the comments in the CSV file 

tsv_file = csv.reader(CommentedFile(open("test.exp", "rb")), 
        delimiter=' ') 


for row in tsv_file: 
    if row != int: 
     next(tsv_file) 
    if row: 
     print row 

코드가 인쇄됩니다 :

['100', '100'] 
['100', '200'] 
['100', '200'] 
['300', '300'] 
Traceback (most recent call last): 
    File "the path", line 57, in <module> 
next(tsv_file) 
StopIteration 

그래서 내가 '여기

# image name 
1.png 
# probe locations 
100 100 
200 100 
100 200 
300 300 

# another image name 
2.png 
100 200 
200 100 
300 300 
135 322 

# end 

내가 함께 연주하고있는 코드입니다 : 여기

은 CSV 파일의 모습입니다 프로그램이 헤더를 기반으로 좌표를 분리하고 별도의 목록에 넣으려고하는 중입니다. 도와 줘서 고마워!

Answer 1

pandas을 살펴보십시오. 그것에는 DataFrame 자료가 들어있어 직관적 인 방법으로 조작 할 수 있습니다. 또한 read_csv 기능을 가지고있어 csv 파일을 다룰 때 많은 번거 로움을 덜어줍니다. 예를 들어

:

import pandas as pd 

#reads your csv file in and returns a DataFrame object as metioned above. 
df = pd.read_csv("your_csv.csv", sep=' ', names=['co_a','co_b'], header=None, skiprows=2) 

#extracts your discordant to separate lists 
list1 = df.co_a.to_list() 
list2 = df.co_b.to_list() 

당신이 당신의 데이터를 관리하여 dataframe과 방식을 볼 수 df 또는 df.head()를 사용할 수 있습니다. df.co_a이 Series 개체라는 것을 언급 할 가치가 있으며, 수퍼 목록/dict를 생각해보십시오. 아마도 거기에서 분석이나 조작을 할 수 있습니다.

또한 내 의견을 csv 파일에 표시하면이를 read_csv으로 무시하는 방법을 보여줄 수 있습니다.

나는 csv module으로 답변을 찾고 있었지만 훨씬 더 발전된 도구이며 장기적으로 도움이 될 것입니다.

희망 하시겠습니까?

Answer 2

코드가 실제로 잘 작동했습니다. 왜 당신이 추적을 받고 있는지 모르겠습니다.

tmp.csv

# image name 
1.png 
# probe locations 
100 100 
200 100 
100 200 
300 300 

# another image name 
2.png 
100 200 
200 100 
300 300 
135 322 

# end 

tmp.py

import csv 

class CommentedFile: 
    def __init__(self, f, commentstring="#"): 
     self.f = f 
     self.commentstring = commentstring 
    def next(self): 
     line = self.f.next() 
     while line.startswith(self.commentstring): 
      line = self.f.next() 
     return line 
    def __iter__(self): 
     return self 

#I did this in order to ignore the comments in the CSV file 

tsv_file = csv.reader(CommentedFile(open("tmp.csv", "rb")), 
        delimiter=' ') 


for row in tsv_file: 
    if row != int: 
     next(tsv_file) 
    if row: 
     print row 

쉘 출력이 사용하는 아마 적절하지 그래서 * 정말 * CSV 파일이 아닙니다

tmp$python tmp.py 
['1.png'] 
['200', '100'] 
['300', '300'] 
['2.png'] 
['200', '100'] 
['135', '322'] 
tmp$uname -mprsv 
Darwin 12.4.0 Darwin Kernel Version 12.4.0: Wed May 1 17:57:12 PDT 2013; root:xnu-2050.24.15~1/RELEASE_X86_64 x86_64 i386 
tmp$python --version 
Python 2.7.2