Как читать столбцы данных .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.

Как мне использовать pandas для анализа этих данных и сохранения их в виде массива python?

Answer 1

Вы можете использовать функцию read_table () в pandas для чтения текстового файла.

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

Затем вы можете использовать встроенную функцию list () для преобразования их в массивы Python.

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