我有一个数据帧，希望根据列中的值复制特定行，并寻找一种简单的方法.
import pandas as pd
data= [['Karan',23],['Rohit',22],['Macron',22], ['Sahil',21],['Aryan',24]]
df = pd.DataFrame(data, columns=['Name','Age'])
因此，我的最终数据帧应该看起来像[行顺序不重要]
Name Age
Karan 23
Rohit 22
Rohit 22
Macron 22
Macron 22
Sahil 21
Aryan 24
Aryan 24
已try [https://stackoverflow.com/questions/24029659/python-pandas-replicate-rows-in-dataframe]但不能处理带有字符串的数据.
您可以使用Index.repeat:
df.loc[df.index.repeat(df['Age'].isin([22,24]).add(1))]
工作原理:
False个值变为1，True变为2)或者，为了获得更大的灵活性，使用numpy.where，您可以 Select 任何想要的值:
import numpy as np
df.loc[df.index.repeat(np.where(df['Age'].isin([22,24]), 2, 1))]
输出:
Name Age
0 Karan 23
1 Rohit 22
1 Rohit 22
2 Macron 22
2 Macron 22
3 Sahil 21
4 Aryan 24
4 Aryan 24
df.loc[df.index.repeat(df['Age'].isin([22,24]).add(1))].reset_index(drop=True)
输出:
Name Age
0 Karan 23
1 Rohit 22
2 Rohit 22
3 Macron 22
4 Macron 22
5 Sahil 21
6 Aryan 24
7 Aryan 24