1. Input data from csv
1.1 normal csv
import csv
with open ("train.csv") as input_file:
data=csv.reader(input_file)
print(data) # only address
raw=[]
for line in data:
raw.append(line)
1.2 pd.read_csv
import pandas as pd
from pandas import DataFrame, Series
with open ("test.csv", encoding="utf-8") as input_file:
data=pd.read_csv(input_file)
print(type(data)) #DataFrame
headers=list(data) #Headers/first row in a list
2. Output a csv
2.1 df output
data_in_df_type.to_csv("path.csv")
2.2 normal csv output from a list
with open("country_info2.csv","w") as out_file:
wr=csv.writer(out_file, delimiter=',')
wr.writerow(header)
for row in rows:
wr.writerow(row)
网友评论