python - python 列表中每个月的工作日

Question

这里是Python新手。我正在尝试创建一个 df，其中 1 列是月份“YYYY-MM”，我在下面的字典中创建了该月份(将不胜感激有关使用代码生成此内容的提示)。我试图使用 Month_list 填充工作日列。

import pandas as pd
df = pd.DataFrame({'month_list':['2019-01', '2019-02', '2019-03', '2019-04']})
df['weekdays'] = ''

我尝试了这段代码，但这仅适用于第一行。 (我明白为什么会这样，我只是不知道正确完成每一行的最佳方法)

df.weekdays = np.busday_count(month_list[0], month_list[0+1])
df

Answer 1

通过 Series.dt.strftime 将值转换为日期时间和字符串，然后添加一个月并传递给 np.busday_count:

df = pd.DataFrame({'month_list':['2019-01', '2019-02', '2019-03', '2019-04']})

month_list = pd.to_datetime(df['month_list'])
a = month_list.dt.strftime('%Y-%m-%d')
b = (month_list + pd.offsets.DateOffset(months=1)).dt.strftime('%Y-%m-%d')

df['weekdays'] = np.busday_count(a, b)

print (df)
  month_list  weekdays
0    2019-01        23
1    2019-02        20
2    2019-03        21
3    2019-04        22

只有 numpy 解决方案:

month_list = pd.to_datetime(df['month_list'])

a = month_list.to_numpy().astype('datetime64[M]')
b = a + np.array([1], dtype='timedelta64[M]') 

df['weekdays'] = np.busday_count(a, b)
print (df)
  month_list  weekdays
0    2019-01        23
1    2019-02        20
2    2019-03        21
3    2019-04        22