Создайте значения столбца Max и Min из одного значения столбца pandas - PullRequest
1 голос
/ 29 мая 2020

У меня есть фрейм данных, подобный приведенному ниже, и мне нужно создать два столбца из базового столбца.

Входные данные

Kg
0.5
0.5
1
1
1
2
2
5
5
5

Ожидается Вывод

Kg_From  Kg_To
0      0.5
0      0.5
0.5    1
0.5    1
0.5    1
1      2
1      2
2      5
2      5
2      5

Как это можно сделать в pandas?

Ответы [ 4 ]

0 голосов
/ 29 мая 2020

Используйте zip и dict для отображения нового столбца, созданного DataFrame.insert, с уникальными отсортированными значениями np.unique с добавлением первого 0 значения np.insert:

df = df.rename(columns={'Kg':'Kg_To'})
a = np.unique(df["Kg_To"])
df.insert(0, 'Kg_from', df['Kg_To'].map(dict(zip(a, np.insert(a, 0, 0)))))
print (df)
   Kg_from  Kg_To
0      0.0    0.5
1      0.0    0.5
2      0.5    1.0
3      0.5    1.0
4      0.5    1.0
5      1.0    2.0
6      1.0    2.0
7      2.0    5.0
8      2.0    5.0
9      2.0    5.0
0 голосов
/ 29 мая 2020

Предполагая, что ваш столбец kg отсортирован:

s = df["Kg"].unique()
df["Kg_from"] = df["Kg"].map({k:v for k,v in zip(s[1:], s)}).fillna(0)
print (df)

    Kg  Kg_from
0  0.5      0.0
1  0.5      0.0
2  1.0      0.5
3  1.0      0.5
4  1.0      0.5
5  2.0      1.0
6  2.0      1.0
7  5.0      2.0
8  5.0      2.0
9  5.0      2.0
0 голосов
/ 29 мая 2020
#get unique values and counts of each value in the Kg column
val,counts = np.unique(df.Kg,return_counts=True)

#shift forward by 1 and replace the first value with 0
val = np.roll(val,1)
val[0] = 0

#repeat the count of each value with the counts generated earlier
df['Kg_from'] = np.repeat(val,counts)

df

     Kg Kg_from
0   0.5 0.0
1   0.5 0.0
2   1.0 0.5
3   1.0 0.5
4   1.0 0.5
5   2.0 1.0
6   2.0 1.0
7   5.0 2.0
8   5.0 2.0
9   5.0 2.0
0 голосов
/ 29 мая 2020

IIU C:

kgs = df.Kg.unique()

lower = [0] + list(kgs[:-1])
kg_dict = {k:v for v,k in zip(lower,kgs)}

# new dataframe
new_df = pd.DataFrame({
             'Kg_From': df['Kg'].map(kg_dict),
             'Kg_To': df['Kg']
         })

# or if you want new columns:
df['Kg_from'] = df['Kg'].map(kg_dict)

Вывод:

   Kg_From  Kg_To
0      0.0    0.5
1      0.0    0.5
2      0.5    1.0
3      0.5    1.0
4      0.5    1.0
5      1.0    2.0
6      1.0    2.0
7      2.0    5.0
8      2.0    5.0
9      2.0    5.0
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...