如何将数据帧从 Pandas 的一个模块导入到另一个模块？

Question

我想重用另一个模块中数据框中的一些列。

我想要类似

file1.py

A   B
25  Hello
30  How are you

file2.py

A   B                 C
25  Hello            Hola
30  How are you      Como estas

我尝试这样做，但它不起作用。

file1.py

import pandas as pd
def cr():
    data = {'state': ['Ohio','Ohio','Ohio','Nevada','Nevada'],
            'year': [2000,2001,2002,2001,2002],
            'pop': [1.5,1.7,3.6,2.4,2.9]}
    df = pd.DataFrame(data)
    return df

输出：

    state   year    pop
0   Ohio    2000    1.5
1   Ohio    2001    1.7
2   Ohio    2002    3.6
3   Nevada  2001    2.4
4   Nevada  2002    2.9

file2.py

from file1 import cr

输出

ImportError: cannot import name 'cr' from 'file1'

Answer 1

其实这看起来不错。如果您尝试以下操作

# file1.py          
def foo():
    print("foo")

# file2.py 
from file1 import foo

foo()

，然后在 file1.py 和 file2.py 旁边的文件夹中运行 python file2.py，程序实际上应该打印foo。

但这仅在您在 file1.py 和 file2.py 所在的文件夹中执行 python 时才有效。

Answer 2

文件1.py

import pandas as pd
def cr():
    data = {'state': ['Ohio','Ohio','Ohio','Nevada','Nevada'],
            'year': [2000,2001,2002,2001,2002],
            'pop': [1.5,1.7,3.6,2.4,2.9]}    
    return data

文件2.py

    import file1
    import pandas as pd
    file1.cr()
    df = pd.DataFrame(file1.cr())