python - 如何使用变量值来调用数据框？

Question

假设我得到了一个总结不同公司的数据框:

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]
.
.

如果我有任何关于如何让它发挥作用的想法，我们将不胜感激。

提前致谢!

Answer 1

您可以使用 dictionary包含您的 DataFrame 并使用字符串作为键。

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.