Как удалить и сдвинуть значения в столбце pandas df - PullRequest
0 голосов
/ 01 июня 2018

У меня есть pandas df, которым я хочу манипулировать, поэтому он упорядочен.Так что для df ниже я бы хотел заказать ['I'].Значения будут читать 10-50.У меня есть 2 варианта сделать это;

1) Попробуйте удалить значения в столбце ['G'] или ['H'].Так что, если значения == X, то удалите.

2) Попробуйте объединить значения в тех же столбцах, когда они == X

import pandas as pd

d = pd.DataFrame({
        'J' : [10,'B','C','C',50],
        'I' : ['B',20,30,40,'C'],
        'H' : ['X','A','C','B','X'],         
        'G' : ['X', 'B', 'A','B','X'],                                 
        })

Вывод:

   G  H   I   J
0  X  X   B  10
1  B  A  20   B
2  A  C  30   C
3  B  B  40   C
4  X  X   C  50

Опция 1 - мы удаляем Xот Column H и предполагаемый вывод будет:

   G  H   I   J
0  X  B  10
1  B  A  20   B
2  A  C  30   C
3  B  B  40   C
4  X  C  50

Вариант 2: мы объединяем X в Column G-H и предполагаемый вывод будет:

   G   H   I   J
0  XX  B  10
1  B   A  20   B
2  A   C  30   C
3  B   B  40   C
4  XX  C  50

Iиграл с df = df.drop(df.H == 'X'), но это удаляет весь ряд.

Ответы [ 4 ]

0 голосов
/ 04 июня 2018

Вы можете пойти на np.where и shift т.е.

ndf = pd.DataFrame(np.where((d['H']=='X')[:,None],
                       d.assign(H=d.H+d.G).shift(-1,axis=1), #only d.shift(...) in case you dont want to add  
                       d), columns=d.columns)

    G  H   I    J
0  XX  B  10  NaN
1   B  A  20    B
2   A  C  30    C
3   B  B  40    C
4  XX  C  50  NaN
0 голосов
/ 03 июня 2018

Более общим решением должно быть изменение условия, если в столбцах G или H указано X, затем объединить вместе и shift по условию:

d = pd.DataFrame({
        'J' : [10,'B','C','C',50, 60],
        'I' : ['B',20,30,40,'C', 'D'],
        'H' : ['X','A','C','B','X', 'Y'],         
        'G' : ['Y', 'B', 'A','B','X', 'X'],                                 
        }, columns=list('GHIJ'))
print (d)
   G  H   I   J
0  Y  X   B  10
1  B  A  20   B
2  A  C  30   C
3  B  B  40   C
4  X  X   C  50
5  X  Y   D  60

m = d[['G','H']].eq('X').any(axis=1)
print (m)

0     True
1    False
2    False
3    False
4     True
5     True
dtype: bool

d['H'] = d['G'] + d['H'] 
d[m] = d[m].shift(-1, axis=1)
print (d)
    G   H   I    J
0  YX   B  10  NaN
1   B  BA  20    B
2   A  AC  30    C
3   B  BB  40    C
4  XX   C  50  NaN
5  XY   D  60  NaN 
0 голосов
/ 04 июня 2018

Для первого вопроса замените 'X' на np.nan, затем отсортируйте фрейм данных (сдвиньте значение)

d.replace({'H':{'X':np.nan}}).apply(lambda x: sorted(x, key=pd.isnull),1).fillna('')
Out[234]:
   G  H   I  J
0  X  B  10
1  B  A  20  B
2  A  C  30  C
3  B  B  40  C
4  X  C  50

Для второго вопроса: сначала с помощью np.where создайте dG,тогда мы делаем то же, что и выше

d.G=np.where((d.G=='X')&(d.H=='X'),'XX',d.G)

d.replace({'H':{'X':np.nan}}).apply(lambda x: sorted(x, key=pd.isnull),1).fillna('')


Out[242]: 
    G  H   I  J
0  XX  B  10   
1   B  A  20  B
2   A  C  30  C
3   B  B  40  C
4  XX  C  50   
0 голосов
/ 03 июня 2018

Опция 1:

Вы можете использовать сдвиг значений влево для строк, которые соответствуют критериям df.H == 'X':

Со следующими определениями переменных:

hij = ['H', 'I', 'J']
x = df.H=='X'

Мы можем записать сменное назначение кратко.

df.loc[x, hij] = df.loc[x, hij].apply(lambda x: x.shift(-1), axis=1)
outputs:
    G   H   I   J
0   X   B   10  NaN
1   B   A   20  B
2   A   C   30  C
3   B   B   40  C
4   X   C   50  NaN

Вариант 2:

Тот же принцип, но необходимы два утверждения.

мы можем объединить H вG

df.loc[x, 'G'] = df.loc[x, 'G'] + df.loc[x, 'H']
# df.loc[x, 'G'] = df.loc[x, ['G, 'H']].sum(axis=1)
# or df.loc[x, ['G', 'H']].apply(np.sum, axis=1)
# or df.loc[x, 'G'] = df.loc[x, ['G', 'H']].apply(lambda x: (x + x.shift(-1))[0], axis=1)

и смещение, как в варианте 1

df.loc[x, hij] = df.loc[x, hij].apply(lambda x: x.shift(-1), axis=1)
final output:
    G   H   I   J
0   XX  B   10  NaN
1   B   A   20  B
2   A   C   30  C
3   B   B   40  C
4   XX  C   50  NaN
...