Python pandas DateFrame新增数据列
Posted 皓月盈江
tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了Python pandas DateFrame新增数据列相关的知识,希望对你有一定的参考价值。
下面代码在Linux下运行
# _*_coding:utf-8_*_
import os
import pandas as pd
# 自定义函数
def func(x):
if x['temperature'] >= 30:
return '高温'
if x['temperature'] < 10:
return '低温'
return '常温'
# Press the green button in the gutter to run the script.
if __name__ == '__main__':
# 从excel文件读取数据
# df = pd.read_excel(os.getcwd() + "/天气.xlsx")
# print(df)
data = 'date': ['2022-08-21', '2022-08-22', '2022-08-23'],
'weather': ["晴", "阴", "晴"],
'temperature': ["30℃", "20℃", "9℃"]
df = pd.DataFrame(data)
print(df)
# 设置索引为日期,方便按日期筛选
df.set_index('date', inplace=True)
print(df)
# 把温度数据去掉'℃',转换成int类型
df.loc[:, 'temperature'] = df['temperature'].str.replace('℃', '').astype(int)
print(df)
# 1.直接赋值,新增'wind'列
df.loc[:, 'wind'] = ['西南风', '东北风', '偏南风']
# 2.使用apply方法,新增'temp_type'列
df.loc[:, 'temp_type'] = df.apply(func, axis=1)
print(df)
# 保存到excel文件
df.to_excel(os.getcwd() + '/天气_新.xlsx')
以上是关于Python pandas DateFrame新增数据列的主要内容,如果未能解决你的问题,请参考以下文章
Python pandas DateFrame查询数据df.loc的5种方法
Python pandas DateFrame查询数据df.loc的5种方法