如何在 Pandas 中将 dataframe 从一个模块导入到另一个模块？

Question

我想重复使用另一个模块中 dataframe 中的一些列。

我想要类似的东西

文件1.py

A   B
25  Hello
30  How are you

文件2.py

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

我试过这样做，但没有用。

文件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]}
    df = pd.DataFrame(data)
    return df

output :

    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

文件2.py

from file1 import cr

output

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())