pandas_根据一列中的值进行行复制

import pandas as pd
import numpy as np

# 创建示例 DataFrame
data = {'building_id': [1, 2, 3, 4],
        'retail': [2, 3, 1, 2]}
df = pd.DataFrame(data)

# 根据 'retail' 列中的值复制 'building_id' 列
result = df.loc[df.index.repeat(df['retail'])].reset_index(drop=True)
result.drop(columns=['retail'], inplace=True)

# 输出结果
print(result)

输入：

复制后：