python csv写入不完整_求大神指点一下CSV读取写入的一点问题

该楼层疑似违规已被系统折叠 隐藏此楼查看此楼

import pandas as pd

import csv

import linecache

data=pd.read_csv('DIAGNOSES_ICD_DATA_TABLE.csv')

print data

disease_data=data.groupby(['ICD9_CODE']).get_group(('412'))

print disease_data

subject_data=disease_data['SUBJECT_ID']

for i in disease_data:

#print list

with open('NOTEEVENTS_DATA_TABLE.csv','rb') as csvfile:

reader=csv.DictReader(csvfile)

for row in reader:

# for i in disease_data['SUBJECT_ID']:

# print i

if row['SUBJECT_ID'] == i['SUBJECT_ID']:

print row

import csv

with open("/home/xu/NOTEEVENTS_DATA_TABLE.csv") as cfile:

reader = csv.DictReader(cfile)

f = open("B.txt", "w")

for row in reader:

if row.get("SUBJECT_ID") == i:

print row.get("TEXT")

f.write(row.get("TEXT") + "\n")