根据列值创建行号

Question

我的 jupyter notebook 中有以下数据集作为输入：

Product   Year    Variable
A         2018        2
A         2019        4
B         2018        2
B         2019        3

我想知道在我的数据集中创建循环或其他东西或排序的最快方法是什么，以便我得到以下输出：

Product   Year    Variable   Row_Num
A         2018        2         1
A         2018        2         2 
A         2019        4         1
A         2019        4         2
A         2019        4         3
A         2019        4         4
B         2018        2         1
B         2018        2         2
and so on...

TL;DR - 基于特定列中的变量，我想创建行。例如，如果变量为 3，我想创建该行的 3 个副本，其中有一列的值为 1、2、3。

我认为我找到的一种方法是首先根据我的变量创建重复项，然后使用类似于 rank() 或 row_number() 的函数来创建我的“row_num”列。如果任何人都可以分享其他可能的方法来做同样的事情，那将会很有帮助。 ????

Answer 1

如果我理解正确，您希望为每一行创建 n 副本，其中 n 的值在其中一列中给出。这是一种方法：

df["new_id"] = df.Variable.apply(lambda x: list(range(x)))
df = df.explode("new_id")

输出：

  Product  Year  Variable new_id
0       A  2018         2      0
0       A  2018         2      1
1       A  2019         4      0
1       A  2019         4      1
1       A  2019         4      2
1       A  2019         4      3
2       B  2018         2      0
2       B  2018         2      1
3       B  2019         3      0
3       B  2019         3      1
3       B  2019         3      2

熊猫解决方案

如果由于某种原因，explode 不可用，因为您使用的是旧版本的 pandas，您可以执行以下操作：

cols = df.columns

def make_df(r):
    d = {k: r[k] for k in cols}
    d["new_var"] = range(r["Variable"])
    res = pd.DataFrame(d)
    return res

dfs = []
for row in df.iterrows():
    dfs.append(make_df(row[1]))
    
pd.concat(dfs)

输出是相同的。