Question

我通常会在SQL和excel中做很多事情，我正在尝试用Pandas做。这里有一些不同的争论问题，结合成一个问题，因为它们都有相同的目标。

我在python中有一个数据框df，有三列：

   |  EventID  |  PictureID  |  Date
0  |  1        |  A          |  2010-01-01
1  |  2        |  A          |  2010-02-01
2  |  3        |  A          |  2010-02-15
3  |  4        |  B          |  2010-01-01
4  |  5        |  C          |  2010-02-01
5  |  6        |  C          |  2010-02-15

EventID是唯一的。尽管PictureID +日期不同，但PictureID并不是唯一的。

予。首先，我想添加一个新列：

df['period'] = the month and year that the event falls into beginning 2010-01.

II。其次，我想将数据“融化”到一些新的数据框中，该数据框计算给定时间段内给定PictureID的事件数。我将只使用两个时期的例子。

   |  PictureID  |  Period  | Count
0  |  A          |  2010-01 | 1
1  |  A          |  2010-02 | 2
2  |  B          |  2010-01 | 1
3  |  C          |  2010-02 | 2

这样我就可以将这个新数据帧堆叠（？）为一个为所有唯一PictureID提供周期计数的东西：

   |  PictureID  |  2010-01 | 2010-02
0  |  A          |  1       | 2
1  |  B          |  1       | 0
2  |  C          |  0       | 2

我的感觉是大熊猫很容易做到这一点，这是正确的吗？

[编辑：删除了困惑的第三部分。]

Answer 1

前两部分你可以做：

>>> df['Period'] = df['Date'].map(lambda d: d.strftime('%Y-%m'))
>>> df
   EventID PictureID                Date   Period
0        1         A 2010-01-01 00:00:00  2010-01
1        2         A 2010-02-01 00:00:00  2010-02
2        3         A 2010-02-15 00:00:00  2010-02
3        4         B 2010-01-01 00:00:00  2010-01
4        5         C 2010-02-01 00:00:00  2010-02
5        6         C 2010-02-15 00:00:00  2010-02
>>> grouped = df[['Period', 'PictureID']].groupby('Period')
>>> grouped['PictureID'].value_counts().unstack(0).fillna(0)
Period  2010-01  2010-02
A             1        2
B             1        0
C             0        2

对于第三部分，要么我没有很好地理解这个问题，要么你没有在例子中发布正确的数字。因为第3行A的计数应为2？并且对于第6行中的C应为1.如果期限为六个月......

无论哪种方式，你都应该这样做：

>>> ts = df.set_index('Date')
>>> ts.resample('6M', ...)

更新：这是一个非常难看的方式，我认为我看到了更好的方法，但我找不到SO问题。但是，这也将完成工作......

def for_half_year(row, data):
    date = row['Date']
    pid = row['PictureID']
    # Do this 6 month checking better
    if '__start' not in data or (date - data['__start']).days > 6*30:
        # Reset values
        for key in data:
            data[key] = 0
        data['__start'] = date
    data[pid] = data.get(pid, -1) + 1
    return data[pid]

df['PastSix'] = df.apply(for_half_year, args=({},), axis=1)

数据框在熊猫中与日期和时期进行争吵

1 个答案: