拆分特定行的值而不复制它们

时间:2019-04-26 10:13:44

标签: python pandas

我正在尝试清理我的csv文件。为此,我必须为“数量”> 1的产品复制行。我是这样的:

file_in = file_in.loc[file_in.index.repeat(file_in.´Quantity)].reset_index(drop=True) 

但是现在我有另一个问题。在我的某些产品的另一栏中,我有序列号,这些序列号从一开始就全部添加到了行中,而当我重复时,它们都获得了所有序列号。 现在,我想拆分这些序列号,以便每一行只有一个。

因此,此:

   Quantity   SerialNumbers       ProductName
1   3           AB1 AB2 AB3            PR
2   3           AB1 AB2 AB3            PR
3   3           AB1 AB2 AB3            PR
4   1             CD                   monitor
5   4            NAN                   mouse
6   4            NAN                   mouse
7   4            NAN                   mouse
8   4            NAN                   mouse
9   4          EF1 EF2 EF3 EF4         ace
10   4          EF1 EF2 EF3 EF4         ace 
11   4          EF1 EF2 EF3 EF4         ace
12   4          EF1 EF2 EF3 EF4         ace
13   2            NAN                 screeen
14   2            NAN                 screeen
15   1            NAN                 flash
...

我需要这样:

   Quantity   SerialNumbers       ProductName
1   3           AB1                    PR
2   3           AB2                    PR
3   3           AB3                    PR
4   1           CD                   monitor
5   4           NAN                   mouse
6   4           NAN                   mouse
7   4           NAN                   mouse
8   4           NAN                   mouse
9   4           EF1                     ace
10   4           EF2                     ace 
11   4           EF3                     ace
12   4           EF4                     ace
13   2          NAN                 screeen
14   2          NAN                 screeen
15   1          NAN                 flash
..

我尝试添加一个新列,在该列中,我将首先从序列号创建列表,然后使用groupby函数编辑。 我尝试使用this,但是不起作用。

file_in['Temp_Split'] = numpy.where((~file_in['Temp_Split'].isna()) & (file_in['Quantity']>1),file_in['Temp_Serial'].str.split() & file_in.groupby(['Position','Quantity'])['Temp_Split'].rank(method='first'), file_in['Temp_split'])

我搜索了类似的问题,但它们都是为了在拆分值的同时复制行。 也许我应该更改代码的第一行,以便在重复执行时检查序列号并对其进行编辑。但是也不确定我该怎么做。

编辑:我不打算在将值拆分为一列的同时复制行,而是尝试删除不需要的一行并添加到下一行。如示例所示。仍然有与开始时相同的行数

2 个答案:

答案 0 :(得分:1)

尝试:

df.SerialNumbers.replace({'NAN':np.nan}, inplace=True)
df['count'] = df.groupby('SerialNumbers').SerialNumbers.cumcount()

f = (lambda x: x['SerialNumbers']
               if x['SerialNumbers'] is np.nan
               else x['SerialNumbers'][x['count']])

df['SerialNumbers'] = df.SerialNumbers.str.split(' ')
df['SerialNumbers'] = df.apply(lambda x: f(x), axis=1)

df.drop(columns='count')

    Quantity SerialNumbers ProductName
0          3           AB1          PR
1          3           AB2          PR
2          3           AB3          PR
3          1            CD     monitor
4          4           NaN       mouse
5          4           NaN       mouse
6          4           NaN       mouse
7          4           NaN       mouse
8          4           EF1         ace
9          4           EF2         ace
10         4           EF3         ace
11         4           EF4         ace
12         2           NaN     screeen
13         2           NaN     screeen
14         1           NaN       flash

我使用Groupby.cumcount查找每个字符串的位置,然后str.split  和apply来选择列表中的项目。

答案 1 :(得分:0)

您可以拆分并拿走第一个

import pandas as pd
df1 = pd.DataFrame({'SerialNumbers': {0: 'aaa aa aa', 1: 'as df', 2: 'fg 67', 3: 'as uy'},'ProductName':{0:80,1:12,2:44,3:61}})

df1['new_col'] = df1['SerialNumbers'].str.split(' ').str[0]
df1 = df1.loc[:,['ProductName','new_col']]
print(df1)