Skip to main content
 首页 » 编程设计

python之在 Pandas 中创建公司周数和年份

2025年05月04日10emanlee

假设我们有以下df

import pandas as pd 
 
data = {'Dates' : ['2018-10-15', '2018-02-01', '2018-04-01']} 
data['Dates'] = pd.to_datetime(data.Dates) 
 
print(df) 
    Dates 
0   2018-10-15 
1   2018-02-01 
2   2018-04-01 

在我现在的公司,我们有一个财务周结构,我通常使用 excel 来计算,我想用 Python 来完成

我使用 DateTime 模块来解决我的情况,如下所示

如果月份是 >= 4 (四月)周数是 1(所以我用 ISO 周数减去 13)

如果月份是 < 4我加了 39。

我对 YEAR 使用相同的逻辑 if >= 4 then year + 1 else YEAR

我想我可以使用一个简单的 for 循环,我可以在我的数据帧上使用它

for x in data.Dates: 
    if x.dt.month >= 4: 
       df['Week'] = x.dt.week - 13 
    else: 
        df['Week'] = x.dt.week + 39 

对于这一年

for x in data.Dates: 
    if x.dt.month >= 4: 
       df['Year'] = FY & x.dt.year + 1 
    else: 
         df['Year'] = FY & x.dt.year 

然而,>= 4两者都抛出公式错误。

File "<ipython-input-38-eadb99fdd9db>", line 4 df.Dates.dt.month > 4: ^ SyntaxError: invalid syntax

但是,如果我这样做

data['Week'] = data.Dates.dt.week

这给出了所有周数,我是否遗漏了一些基本或必要的东西?

我希望这是清晰简洁的,任何建议(甚至如何提出更好的问题)我们都将不胜感激。

请您参考如下方法:

不要使用显式循环

Pandas 擅长向量化运算。不需要 for 循环。例如,您可以使用 numpy.where有条件地创建一个系列:

import numpy as np 
 
data['Week'] = np.where(data['Dates'].dt.month >= 4, data['Dates'].dt.week - 13, 
                        data['Dates'].dt.week + 39) 

您的代码不起作用的原因是因为您在每个循环中更新整个系列而不是系列中的元素。换句话说,您正在将逐元素逻辑应用于一个系列。