我爱编程

Pandas DataFrame I/O

2018-04-27  本文已影响16人  Jacaranda2016

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)

上一篇 下一篇

猜你喜欢

热点阅读