Python 3.4, ошибка при создании DataFrame с пандой

Question

Я пытаюсь создать DataFrame со следующим:

from pandas import DataFrame, read_csv 

import matplotlib.pyplot as plt 
import pandas as pd 
import sys 

# The inital set of baby names and birth rates 
names =['Bob','Jessica','Mary','John','Mel'] 
births = [968, 155, 77, 578, 973] 

#Now we wil zip them together 
BabyDataSet = zip(names,births) 
    ##we have to add the 'list' for version 3.x 
print (list(BabyDataSet)) 

#create the DataFrame 
df = DataFrame(BabyDataSet, columns = ['Names', 'Births']) 
print (df) 

, когда я запускаю программу, я получаю следующее сообщение об ошибке: «тип данных не может быть итератор ' Я читал следующее:' What does the "yield" keyword do in Python? ', но я не понимаю, как это относится к тому, что я делаю. Будем очень благодарны за любую помощь и дальнейшее понимание.

Answer 1

В python 3, zip возвращает итератор, а не такой список, как в python 2. Просто преобразуйте его в список при построении DataFrame, как это.

df = DataFrame(list(BabyDataSet), columns = ['Names', 'Births']) 

Answer 2

Вы также можете создать dataframe, используя альтернативный синтаксис, который полностью исключает проблему с zip-генератором.

df = DataFrame({'Names': names, 'Births': births}) 

Читать documentation на инициализирующих dataframes. Pandas просто берет словарь, создает по одному столбцу для каждой записи с ключом как имя и значение в качестве значения.

Dict can contain Series, arrays, constants, or list-like objects