如何用变量值调用数据框?

3 投票
1 回答
5652 浏览
提问于 2025-04-18 04:51

假设我有一个数据表,里面总结了不同公司的信息:

summary=pandas.DataFrame(columns=['Company Name', 'Formation Date', 'Revenue', 'Profit', 'Loss'])

然后,每个公司在这个数据表中都有一个对应的数据表,名字就是公司的名字,这个数据表提供了更详细的公司历史和统计数据。比如说:

exampleco=pandas.Dataframe(columns=['Date', 'Daily Profit', 'Daily Loss', 'Daily Revenue'])

我有一个脚本,可以处理这个summary数据表中的每一行,但我想从row['Company Name']中提取公司的名字,然后用这个名字去访问公司的那个专属数据表。

换句话说,我希望有一种方法可以像这样操作:

.
.
>>> company=row['Company Name']
>>> pandas.get_dataframe_from_variable(company)

Empty DataFrame
Columns: ['Date', 'Daily Profit', 'Daily Loss', 'Daily Revenue']
Index: []

[0 rows x 2 columns]
.
.

如果你有什么想法能让我实现这个功能,我会非常感激。

提前谢谢你们!

1 个回答

1

你可以用一个字典来存放你的数据框(DataFrames),并用字符串作为它的键。

companies = {'company1':pandas.DataFrame(columns=['Date', 'Daily Profit', 
                                                 'Daily Loss', 'Daily Revenue']), 
             'company2':pandas.DataFrame(columns=['Date', 'Daily Profit',
                                                 'Daily Loss', 'Daily Revenue'])}

company=row['Company Name'] # Get your company name as a string from your summary.
company_details = companies[company] # Returns a DataFrame.

撰写回答