如何使用变量值来调用数据帧?

时间:2014-04-30 09:20:43

标签: python variables pandas dataframe

假设我有一个汇总不同公司的数据框:

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 个答案:

答案 0 :(得分:1)

您可以使用dictionary来包含数据框,并使用字符串作为键。

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.