python - 使用 numpy 重复函数创建 pandas DataFrame

Question

我正在尝试创建一个 pandas DataFrame，其中一列多次使用 numpy repeat 函数。这是我在 R 中使用 c 和 rep 执行此操作的方法，它有效:

df <- data.frame(
  date = seq.Date(as.Date('2018-12-01'), as.Date('2019-12-01'), by='month'),
  value = c(rep(0.08, 7), rep(0.06, 6)),
)

这是我在 pandas 中尝试的方法，但它抛出错误 arrays must all be same length:

import numpy as np
import pandas as pd
df= pd.DataFrame({
    'date': pd.date_range('2018-12-01', '2019-12-01', freq='MS'),
    'value': [np.repeat(0.08, 7), np.repeat(0.06, 6)]
})

我如何在 pandas 中执行此操作？

Answer 1

np.repeat 可以为 repeats 参数采用类似数组的方式。

df = pd.DataFrame({
    'date': pd.date_range('2018-12-01', '2019-12-01', freq='MS'),
    'value': np.repeat([.08, .06], [7, 6])
})