Python pandas DateFrame新增数据列

Posted 2022-08-30 皓月盈江

下面代码在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')