gpt4 book ai didi

python pandas 插入列

转载 作者:太空狗 更新时间:2023-10-30 02:29:17 25 4
gpt4 key购买 nike

我正在编写代码以在 csv 文件中插入一个新列:

import sys,os,csv,glob
dir = os.path.dirname(__file__)

import pandas as pd

updatecsv()

def updatecsv():

files = 'example.cs'
df = pd.read_csv(files)
df = df.convert_objects(convert_numeric=True)
#until here, the code is running fine
#now i wanted to add a new column in a specific index with all value =10
df.insert(2,'new',1000)

当我运行代码时,没有报错。当我打开 csv 文件时,没有添加新行。我决定使用 python shell 进行检查:

>>>files = 'example.csv'
>>>df = pd.read_csv(files)
>>>df = df.convert_objects(convert_numeric=True)
>>>df
A B C D
0 1 2 3 4
1 5 6 7 8
2 9 10 11 12
df['new']=13
>>>df
A B C D new
0 1 2 3 4 13
1 5 6 7 8 13
2 9 10 11 12 13
>>>df['new'] = df['new'] +1
>>>df
A B C D new
0 1 2 3 4 14
1 5 6 7 8 14
2 9 10 11 12 14
>>>df.insert(2,'win',22)
>>>df
A B win C D new
0 1 2 22 3 4 14
1 5 6 22 7 8 14
2 9 10 22 11 12 14

使用 python shell,我只能在 shell 上看到更新的结果。如何在 CSV 文件中更新它?

最佳答案

当你这样做时——

df.insert(2,'new',1000)

它将 new 列插入内存中的 DataFrame df(所有值为 1000)。它不会自动将其写回 csv。

对于要写回 csv 的数据帧所做的更改,您应该使用 DataFrame.to_csv()方法。示例 -

def updatecsv():
files = 'example.cs'
df = pd.read_csv(files)
df = df.convert_objects(convert_numeric=True)
#until here, the code is running fine
#now i wanted to add a new column in a specific index with all value =10
df.insert(2,'new',1000)
df.to_csv(files)

此外,您应该确保在尝试调用之前定义函数。

关于python pandas 插入列,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/33139513/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com