R中的工作代码
library(dplyr)
tmp <- test %>%
group_by(InvoiceDocNumber) %>%
summarise(invoiceprob=max(itemprob)) %>%
mutate(invoicerank=rank(desc(invoiceprob)))
但是我想用python重写代码。我写了下面的代码,但它抛出了错误。我正在使用python中可用的类似版本的dplyr。
from dfply import *
tmp = (test >>
group_by(test.InvoiceDocNumber) >>
summarize(invoiceprob=max(test.itemprob)) >>
mutate(invoicerank=rankdata(test.invoiceprob)))
AttributeError: 'DataFrame' object has no attribute 'invoiceprob'
谁能帮我 ?
您可以使用assign
它来一个链:
(
test.groupby("InvoiceDocNumber", as_index=False)
.itemprob.max()
.rename(columns={"itemprob":"invoiceprob"})
.assign(invoicerank = lambda x: x.invoiceprob.rank(ascending=False))
)
输出:
InvoiceDocNumber invoiceprob invoicerank
0 0 0.924193 5.0
1 1 0.974173 4.0
2 2 0.978962 3.0
3 3 0.992663 2.0
4 4 0.994243 1.0
数据:
import numpy as np
import pandas as pd
n = 100
test = pd.DataFrame({"InvoiceDocNumber": np.random.choice(np.arange(5), size=n),
"itemprob": np.random.uniform(size=n)})
本文收集自互联网,转载请注明来源。
如有侵权,请联系 [email protected] 删除。
我来说两句