python - 将 transform 与 nth 一起使用

Question

我正在使用以下代码添加一个带有 transform 的列:

df['new_date'] = df.groupby('account')['date'].transform('last')

这工作正常，但默认情况下它会删除 NaN(如现有错误 here、here 和 here 中所述)，我想保留它。开发者建议改用 nth(-1)。没问题!

但是，我不知道如何将它与transform 一起使用。错误信息为

`df.groupby('a')['b'].transform('nth')`

is nth() missing 1 required positional argument: 'n'，这似乎很诱人地暗示 transform 识别该方法，只要我能想出一种方法将索引传递给它.但是没有一个

df.groupby('a')['b'].transform('nth(-1)')
df.groupby('a')['b'].transform('nth'(-1))
df.groupby('a')['b'].transform('nth')(-1)

工作。有什么方法可以做到这一点吗？

Answer 1

此处可以使用第二个参数传递给 GroupBy.nth 的值:

np.random.seed(2015)

df = pd.DataFrame({'account': ['foo', 'bar', 'baz'] * 3,
                   'val': np.random.choice([np.nan, 1],size=9)})
#print (df)

df['val1'] = df.groupby('account')['val'].transform('last')
df['val2'] = df.groupby('account')['val'].transform('nth', -1)
print (df)
  account  val  val1  val2
0     foo  NaN   1.0   1.0
1     bar  NaN   1.0   NaN
2     baz  NaN   NaN   NaN
3     foo  NaN   1.0   1.0
4     bar  1.0   1.0   NaN
5     baz  NaN   NaN   NaN
6     foo  1.0   1.0   1.0
7     bar  NaN   1.0   NaN
8     baz  NaN   NaN   NaN