Pandas讀寫CSV文件的方法示例
讀csv
使用pandas讀取
import pandas as pd import csv if name == '__main__': # header=0——表示csv文件的第一行默認(rèn)為dataframe數(shù)據(jù)的行名稱, # index_col=0——表示使用第0列作為dataframe的行索引, # squeeze=True——表示如果文件只包含一列,則返回一個序列。 file_dataframe = pd.read_csv('../datasets/data_new_2/csv_file_name.csv', header=0, index_col=0, squeeze=True) # 結(jié)果:
# 當(dāng)參數(shù)index_col=False 時,自動生成行索引0到n
# csv數(shù)據(jù):
data_1 = [] # 讀取行索引一樣的數(shù)據(jù),保存為list try: # 行索引為i的數(shù)據(jù)有多行,列為'pre_star' data_1.extend(file_dataframe .loc[i]['pre_star'].values.astype(float)) except AttributeError: # 行索引為i的數(shù)據(jù)只有單行, data_1.extend([file_dataframe .loc[i]['pre_star']]) # 多行結(jié)果
# 行索引為i的數(shù)據(jù)只有一行時,不能對file_dataframe .loc[i]['pre_star']使用.values,否則會報錯:
寫csv
使用csv寫
stu1 = [lid, k, pre_count_data[k]] # 打開文件,寫模式為追加'a' out = open('../results/write_file.csv', 'a', newline='') # 設(shè)定寫入模式 csv_write = csv.writer(out, dialect='excel') # 寫入具體內(nèi)容 csv_write.writerow(stu1)
以上就是本文的全部內(nèi)容,希望對大家的學(xué)習(xí)有所幫助,也希望大家多多支持腳本之家。
相關(guān)文章
VSCode搭建Django開發(fā)環(huán)境的圖文步驟
本篇介紹在vscode環(huán)境下搭建Django開發(fā)環(huán)境的詳細(xì)步驟,包括Python、Django、VSCode等,以及它們的安裝和配置方法,具有一定的參考價值,感興趣的可以了解一下2023-09-09python中py文件與pyc文件相互轉(zhuǎn)換的方法實例
pyc是一種二進(jìn)制文件,是由py文件經(jīng)過編譯后,生成的文件,下面這篇文章主要給大家介紹了關(guān)于python中py文件與pyc文件相互轉(zhuǎn)換的相關(guān)資料,需要的朋友可以參考下2022-05-05Blender Python編程創(chuàng)建發(fā)光材質(zhì)示例詳解
這篇文章主要為大家介紹了Blender Python編程創(chuàng)建發(fā)光材質(zhì)示例詳解,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進(jìn)步,早日升職加薪2022-08-08python使用matplotlib定制繪圖的線型、標(biāo)記類型
這篇文章主要給大家詳細(xì)介紹了python使用matplotlib定制繪圖的線型、標(biāo)記類型,文中有詳細(xì)的代碼示例,具有一定的參考價值,需要的朋友可以參考下2023-07-07