python - Pandas 生成柱

Question

我有以下 df:

from    to  

5   7   

8   10  

我正在尝试获得以下输出:

from    to  zip_1   zip_2   zip_3   zip_4

5   8   5   6   7   8

9   10  9   10      

我尝试了以下代码，但导致错误。

for index, row in df.iterrows():
    zip_index = 0
    colName = 'zip' + '_' + str(zip_index)
    while row[colName] != row.To: 
        zip_index = zip_index + 1
        NewCol = 'zip' + '_' + str(zip_index)
        row[NewCol] = row[colName] + 1
        colName = NewCol

我错过了什么？
谢谢!

Answer 1

您可以使用 pandas.DataFrame.apply :

df2 = df.apply(lambda x: pd.Series(range(x["from"], x["to"] + 1), dtype = "float16"), axis = 1)

df2.columns += 1
df2 = df2.add_prefix("zip_")

df_final = pd.concat([df,df2], axis = 1)
#   from  to  zip_1  zip_2  zip_3  zip_4
#0     5   8    5.0    6.0    7.0    8.0
#1     9  10    9.0   10.0    NaN    NaN