pandas處理csv文件的方法步驟
一、我的需求
對(duì)于這樣的一個(gè) csv 表,需要將其
(1)將營(yíng)業(yè)部名稱和日期和股票代碼進(jìn)行拼接
(2)對(duì)于除了買入金額不同的的數(shù)據(jù)需要將它們的買入金額相加,每個(gè)買入金額乘以買賣序號(hào)的符號(hào)表示該營(yíng)業(yè)名稱對(duì)應(yīng)的買入金額
比如:xx公司,20190731,1,股票1,4000,C20201010,xxxx
我這里想要的結(jié)果是:xx公司2019713C20201010,4000

二、代碼
(1)首先由于文件是 gbk,所以讀取是需要注意 encoding
(2)日期是int類型,所以需要轉(zhuǎn)化為 字符串
import pandas as pd
import numpy as np
#讀取數(shù)據(jù)
filename = "test.csv"
# 讀取 excel 表,根據(jù)文件的編碼指定編碼方式
data = pd.read_csv(filename, encoding='gbk')
# 將所有內(nèi)容轉(zhuǎn)為字符串
# data = data.applymap(str)
# 將日期這一列轉(zhuǎn)為字符串
data['日期'] = data['日期'].apply(str)
# print(data.loc[0,'營(yíng)業(yè)部名稱'])
# print(data.loc[0,'日期'])
# print(data.loc[0,'股票代碼'])
# print(data.loc[0,'買賣序號(hào)'])
# print(data.loc[0,'買入金額'])
# 拼接:營(yíng)業(yè)部名稱+日期+股票代碼
data['name_date_code'] = data['營(yíng)業(yè)部名稱'] + data['日期'] + data['股票代碼']
# 取"買賣序號(hào)"的符號(hào)和買入金額相乘
# np.sign 獲取序號(hào)對(duì)應(yīng)的符號(hào)
data['buy'] = np.sign(data['買賣序號(hào)']) * data['買入金額']
data = data.drop(['營(yíng)業(yè)部名稱', '日期', '買賣序號(hào)', '股票名', '買入金額', '股票代碼', 'data_stock'], axis=1)
# 將 name_date_code 相同的行,金額相加
buy_sum = data.groupby('name_date_code')['buy'].sum()
# 將相加的金額加入數(shù)據(jù)data,缺失數(shù)據(jù)用0填充
data['buy_sum'] = data.loc[:, 'name_date_code'].map(buy_sum).fillna(0)
# 將買入金額刪掉,只剩下兩列數(shù)據(jù)
data = data.drop(['buy'], axis=1)
# 刪除重復(fù)行
data = data.drop_duplicates()
# 寫入數(shù)據(jù),同樣需要注意指定編碼格式
data.to_csv("YYBD_result.csv", encoding='gbk',index=False)
三、總結(jié)
(1)編碼格式,正常是 utf-8 的不用指定,用默認(rèn)的即可
(2)pandas 讀取一行數(shù)據(jù)
# data.iloc 取一整行 print(data.iloc[0])
(3)pandas 處理數(shù)據(jù)確實(shí)很厲害,字符串拼接,類型轉(zhuǎn)換,刪除重復(fù)行,真方便
到此這篇關(guān)于pandas處理csv文件的方法步驟的文章就介紹到這了,更多相關(guān)pandas處理csv文件內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家!
相關(guān)文章
python實(shí)現(xiàn)字符串和數(shù)字拼接
今天小編就為大家分享一篇python實(shí)現(xiàn)字符串和數(shù)字拼接,具有很好的參考價(jià)值,希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2020-03-03
Python爬蟲(chóng)包BeautifulSoup實(shí)例(三)
這篇文章主要為大家詳細(xì)介紹了Python爬蟲(chóng)包BeautifulSoup實(shí)例,具有一定的參考價(jià)值,感興趣的朋友可以參考一下2018-06-06
使用Python圖像處理庫(kù)Pillow處理圖像文件的案例分析
本文將通過(guò)使用Python圖像處理庫(kù)Pillow,幫助大家進(jìn)一步了解Python的基本概念:模塊、對(duì)象、方法和函數(shù)的使用,文中代碼講解的非常詳細(xì),需要的朋友可以參考下2023-07-07
python輸出100以內(nèi)的質(zhì)數(shù)與合數(shù)實(shí)例代碼
本文通過(guò)實(shí)例代碼給大家介紹了python輸出100以內(nèi)的質(zhì)數(shù)與合數(shù)的方法,非常不錯(cuò),具有一定的參考借鑒價(jià)值,需要的朋友參考下吧2018-07-07
python的getattr和getattribute攔截內(nèi)置操作實(shí)現(xiàn)
在Python中,getattr和getattribute是用于動(dòng)態(tài)屬性訪問(wèn)和自定義屬性訪問(wèn)行為的重要工具,本文主要介紹了python的getattr和getattribute攔截內(nèi)置操作實(shí)現(xiàn),具有一定的參考價(jià)值,感興趣的可以了解一下2024-01-01
python按順序重命名文件并分類轉(zhuǎn)移到各個(gè)文件夾中的實(shí)現(xiàn)代碼
這篇文章主要介紹了python按順序重命名文件并分類轉(zhuǎn)移到各個(gè)文件夾中,本文通過(guò)實(shí)例代碼給大家介紹的非常詳細(xì),對(duì)大家的學(xué)習(xí)或工作具有一定的參考借鑒價(jià)值,需要的朋友可以參考下2020-07-07
使用PyOpenGL繪制三維坐標(biāo)系實(shí)例
今天小編就為大家分享一篇使用PyOpenGL繪制三維坐標(biāo)系實(shí)例,具有很好的參考價(jià)值,希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2019-12-12
Python中Turtle庫(kù)改變畫筆(海龜)方向的兩種方法總結(jié)
turtle庫(kù)是python標(biāo)準(zhǔn)庫(kù)之一,入門級(jí)繪圖庫(kù),import turtle之后即可使用,下面這篇文章主要給大家介紹了關(guān)于Python中Turtle庫(kù)改變畫筆(海龜)方向的兩種方法,需要的朋友可以參考下2022-11-11
對(duì)python 操作solr索引數(shù)據(jù)的實(shí)例詳解
今天小編就為大家分享一篇對(duì)python 操作solr索引數(shù)據(jù)的實(shí)例詳解,具有很好的參考價(jià)值,希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2018-12-12

