如何使用熊猫读取.txt数据列并将其馈送到TF

Question

我有一个文件birth_life_data.txt，格式为：

Country fertiliy_rate Life_span 
India 5.7 52
USA 10.2 70
...
...

我想将fertiliy_rate数据存储在变量X中，并将life_span类似地在Y中存储为1-d数组，以作为训练数据输入到我的TF图中。

我如何使用熊猫解析这些数据并将其以python数组的形式存储？

Answer 1

您可以在熊猫中使用read_table()函数来读取文本文件。

df = pd.read_table('birth_life_data.txt', delim_whitespace=True)

然后您可以使用内置的list（）函数将它们转换为python数组。

X = list(df['Country'])
Y = list(df['Life_span'])