如何使用列表推导式将pandas DataFrame两列转为元组列表
可以通过列表推导式实现该转换,具体操作如下:
首先确认你的DataFrame中dates列的格式:如果从CSV导入后dates为datetime类型,需要先转换为YYYY/MM/DD格式的字符串,若已经是目标格式的字符串可跳过该步骤:
df['dates'] = df['dates'].dt.strftime('%Y/%m/%d')
然后使用列表推导式遍历DataFrame的每行,生成目标格式的列表:
cashflow_list = [(row['dates'], row['cashflows']) for _, row in df.iterrows()]
如果你的数据集规模较大,更推荐使用pandas内置方法实现,性能优于列表推导式写法:
cashflow_list = list(df.itertuples(index=False, name=None))
两种写法最终得到的cashflow_list就是你要求的输出格式。
内容的提问来源于stack exchange,提问作者Camilla
相关产品推荐
相关产品推荐

