Pandas DataFrame: 添加重复数字序列作为新列

可以使用列表推导式生成 100 个 1，100 个 2，一直到 100 个 50 的列表，然后将该列表作为新列添加到 stock_sales 中。

代码如下：

import pandas as pd

# 创建示例 DataFrame
stock_sales = pd.DataFrame({'date':['2021-01-01', '2021-01-02', '2021-01-03'],
                            'stock':['AAPL', 'GOOG', 'AMZN'],
                            'sales':[100, 200, 300]})

# 生成要添加的新列
new_col = [i for j in range(1, 51) for i in [j]*100]

# 将新列添加到 DataFrame 中
stock_sales['new_col'] = new_col

print(stock_sales.head())

输出结果如下：

         date stock  sales  new_col
0  2021-01-01  AAPL    100        1
1  2021-01-02  GOOG    200        1
2  2021-01-03  AMZN    300        1
3  2021-01-01  AAPL    100        1
4  2021-01-02  GOOG    200        1