Розділити (вибухнути) запис рядка даних панд даних на окремі рядки


200

У мене є a, pandas dataframeв якому один стовпець текстових рядків містить знаки, розділені комами. Я хочу розділити кожне поле CSV та створити новий рядок на запис (припустимо, що CSV чистий і його потрібно розділити лише на ','). Наприклад, aповинні стати b:

In [7]: a
Out[7]: 
    var1  var2
0  a,b,c     1
1  d,e,f     2

In [8]: b
Out[8]: 
  var1  var2
0    a     1
1    b     1
2    c     1
3    d     2
4    e     2
5    f     2

Поки я пробував різні прості функції, але .applyметод, здається, приймає лише один рядок як повернене значення, коли він використовується на осі, і я не можу отримати.transform до роботи. Будь-які пропозиції були б вдячні!

Приклад даних:

from pandas import DataFrame
import numpy as np
a = DataFrame([{'var1': 'a,b,c', 'var2': 1},
               {'var1': 'd,e,f', 'var2': 2}])
b = DataFrame([{'var1': 'a', 'var2': 1},
               {'var1': 'b', 'var2': 1},
               {'var1': 'c', 'var2': 1},
               {'var1': 'd', 'var2': 2},
               {'var1': 'e', 'var2': 2},
               {'var1': 'f', 'var2': 2}])

Я знаю, що це не спрацює, оскільки ми втрачаємо метадані DataFrame, перебираючи numpy, але це повинно дати вам уявлення про те, що я намагався зробити:

def fun(row):
    letters = row['var1']
    letters = letters.split(',')
    out = np.array([row] * len(letters))
    out['var1'] = letters
a['idx'] = range(a.shape[0])
z = a.groupby('idx')
z.transform(fun)

2
Інші рішення на цій сторінці працюють, але я знайшов наступне коротке та ефективне. stackoverflow.com/questions/27263805 / ...
desaiankitb

1
Для інших, хто приходить на цю сторінку та шукає рішення, яке зберігає кілька стовпців, подивіться на це питання: stackoverflow.com/questions/17116814/…
Sos

Відповіді:


81

Як щодо щось подібне:

In [55]: pd.concat([Series(row['var2'], row['var1'].split(','))              
                    for _, row in a.iterrows()]).reset_index()
Out[55]: 
  index  0
0     a  1
1     b  1
2     c  1
3     d  2
4     e  2
5     f  2

Тоді потрібно просто перейменувати стовпці


1
Схоже, це вийде. Спасибі за вашу допомогу! В цілому, однак, чи є кращий підхід до Split-Apply-Combine, де Apply повертає фрейм даних довільного розміру (але відповідає всім фрагментам), а Combine просто vstacks повернених DF?
Вінсент

GroupBy.apply повинен працювати (я просто спробував це проти майстра). Однак у цьому випадку вам насправді не потрібно пройти додатковий крок групування, оскільки ви генеруєте дані за рядками, чи не так?
Chang She

1
Гей, хлопці. Вибачте, що заскочив у це так пізно, але цікаво, чи немає кращого рішення для цього. Я намагаюся експериментувати з iterrow вперше, оскільки це здається квитком на це. Мене також бентежить запропоноване рішення. Що означає "_"? Чи можете ви пояснити, як працює рішення? --Thank вас
horatio1701d

11
Чи можна розширити рішення на більше двох стовпців?
horatio1701d

1
будь ласка, перевірте цей
векторний

146

UPDATE2: більш загальна векторна функція, яка буде працювати для декількох normalі декількох listстовпців

def explode(df, lst_cols, fill_value='', preserve_index=False):
    # make sure `lst_cols` is list-alike
    if (lst_cols is not None
        and len(lst_cols) > 0
        and not isinstance(lst_cols, (list, tuple, np.ndarray, pd.Series))):
        lst_cols = [lst_cols]
    # all columns except `lst_cols`
    idx_cols = df.columns.difference(lst_cols)
    # calculate lengths of lists
    lens = df[lst_cols[0]].str.len()
    # preserve original index values    
    idx = np.repeat(df.index.values, lens)
    # create "exploded" DF
    res = (pd.DataFrame({
                col:np.repeat(df[col].values, lens)
                for col in idx_cols},
                index=idx)
             .assign(**{col:np.concatenate(df.loc[lens>0, col].values)
                            for col in lst_cols}))
    # append those rows that have empty lists
    if (lens == 0).any():
        # at least one list in cells is empty
        res = (res.append(df.loc[lens==0, idx_cols], sort=False)
                  .fillna(fill_value))
    # revert the original index order
    res = res.sort_index()
    # reset index if requested
    if not preserve_index:        
        res = res.reset_index(drop=True)
    return res

Демонстрація:

Кілька listстовпців - усі listстовпці повинні мати однакове число елементів у кожному рядку:

In [134]: df
Out[134]:
   aaa  myid        num          text
0   10     1  [1, 2, 3]  [aa, bb, cc]
1   11     2         []            []
2   12     3     [1, 2]      [cc, dd]
3   13     4         []            []

In [135]: explode(df, ['num','text'], fill_value='')
Out[135]:
   aaa  myid num text
0   10     1   1   aa
1   10     1   2   bb
2   10     1   3   cc
3   11     2
4   12     3   1   cc
5   12     3   2   dd
6   13     4

збереження початкових значень індексу:

In [136]: explode(df, ['num','text'], fill_value='', preserve_index=True)
Out[136]:
   aaa  myid num text
0   10     1   1   aa
0   10     1   2   bb
0   10     1   3   cc
1   11     2
2   12     3   1   cc
2   12     3   2   dd
3   13     4

Налаштування:

df = pd.DataFrame({
 'aaa': {0: 10, 1: 11, 2: 12, 3: 13},
 'myid': {0: 1, 1: 2, 2: 3, 3: 4},
 'num': {0: [1, 2, 3], 1: [], 2: [1, 2], 3: []},
 'text': {0: ['aa', 'bb', 'cc'], 1: [], 2: ['cc', 'dd'], 3: []}
})

Стовпець CSV:

In [46]: df
Out[46]:
        var1  var2 var3
0      a,b,c     1   XX
1  d,e,f,x,y     2   ZZ

In [47]: explode(df.assign(var1=df.var1.str.split(',')), 'var1')
Out[47]:
  var1  var2 var3
0    a     1   XX
1    b     1   XX
2    c     1   XX
3    d     2   ZZ
4    e     2   ZZ
5    f     2   ZZ
6    x     2   ZZ
7    y     2   ZZ

використовуючи цей маленький трюк, ми можемо перетворити стовпчик, схожий на CSV, у listстовпчик:

In [48]: df.assign(var1=df.var1.str.split(','))
Out[48]:
              var1  var2 var3
0        [a, b, c]     1   XX
1  [d, e, f, x, y]     2   ZZ

ОНОВЛЕННЯ: загальний векторний підхід (працюватиме також для кількох стовпців):

Оригінальний DF:

In [177]: df
Out[177]:
        var1  var2 var3
0      a,b,c     1   XX
1  d,e,f,x,y     2   ZZ

Рішення:

спочатку давайте перетворимо рядки CSV у списки:

In [178]: lst_col = 'var1' 

In [179]: x = df.assign(**{lst_col:df[lst_col].str.split(',')})

In [180]: x
Out[180]:
              var1  var2 var3
0        [a, b, c]     1   XX
1  [d, e, f, x, y]     2   ZZ

Тепер ми можемо зробити це:

In [181]: pd.DataFrame({
     ...:     col:np.repeat(x[col].values, x[lst_col].str.len())
     ...:     for col in x.columns.difference([lst_col])
     ...: }).assign(**{lst_col:np.concatenate(x[lst_col].values)})[x.columns.tolist()]
     ...:
Out[181]:
  var1  var2 var3
0    a     1   XX
1    b     1   XX
2    c     1   XX
3    d     2   ZZ
4    e     2   ZZ
5    f     2   ZZ
6    x     2   ZZ
7    y     2   ZZ

СТАРА відповідь:

Натхненний рішенням @AFinkelstein , я хотів зробити його більш узагальненим, що можна було б застосувати до DF з більш ніж двома стовпцями і так само швидко, ну майже так само швидко, як і рішення AFinkelstein):

In [2]: df = pd.DataFrame(
   ...:    [{'var1': 'a,b,c', 'var2': 1, 'var3': 'XX'},
   ...:     {'var1': 'd,e,f,x,y', 'var2': 2, 'var3': 'ZZ'}]
   ...: )

In [3]: df
Out[3]:
        var1  var2 var3
0      a,b,c     1   XX
1  d,e,f,x,y     2   ZZ

In [4]: (df.set_index(df.columns.drop('var1',1).tolist())
   ...:    .var1.str.split(',', expand=True)
   ...:    .stack()
   ...:    .reset_index()
   ...:    .rename(columns={0:'var1'})
   ...:    .loc[:, df.columns]
   ...: )
Out[4]:
  var1  var2 var3
0    a     1   XX
1    b     1   XX
2    c     1   XX
3    d     2   ZZ
4    e     2   ZZ
5    f     2   ZZ
6    x     2   ZZ
7    y     2   ZZ

7
чувак, якщо ти можеш відкрити дискусію в пандах Git, я думаю, нам потрібна така функція, як ця !!! Я бачив стільки запитань щодо нерозпізнання та непотрібності в ПП для панд
YOBEN_S

як це використовувати для кількох стовпців. Як, якщо у мене є дані, розділені комами в 2 стовпцях, і я хочу це робити послідовно?
Джаскаран Сінгх Пурі

@JaskaranSinghPuri, ви хочете спочатку перетворити всі стовпці CSV у списки.
MaxU

1
Наразі це не працює, якщо елементи вашого списку є кортежами. Але після перетворення всього кордону в рядок це працює як шарм!
Гвідо

2
Схоже, прохання Венбена почули боги панди, вони встановили .explode()метод в API (також дивіться цю відповідь ).
cs95

117

Після болісного експерименту, щоб знайти щось швидше, ніж прийнята відповідь, я змусив це працювати. Він набрав приблизно 100 разів швидше на наборі даних, на якому я його спробував.

Якщо хтось знає спосіб зробити це більш елегантним, будь ласка, змініть мій код. Я не зміг знайти спосіб, який працює без встановлення інших стовпців, які ви хочете зберегти як індекс, а потім скидання індексу та перейменування стовпців, але я б уявив, що працює щось інше.

b = DataFrame(a.var1.str.split(',').tolist(), index=a.var2).stack()
b = b.reset_index()[[0, 'var2']] # var1 variable is currently labeled 0
b.columns = ['var1', 'var2'] # renaming var1

2
Це рішення працювало значно швидше і, здається, використовує менше пам’яті,
Кирило

1
Це приємне векторне рішення панди, я це шукав. Дякую!
Денніс Голомазов

Коли я спробую це на власному наборі даних, я продовжую отримувати TypeError: object of type 'float' has no len()на першому кроці ( DataFrame(df.var1.str.split(',').tolist()))
user5359531

@ user5359531 Ваш набір даних, ймовірно, містить NaNу цьому стовпці, тому замінаb = DataFrame(a.var1.str.split(',').values.tolist(), index=a.var2).stack()
Flair

Тільки фій, ось приємно написати це рішення із прикладом.
hhbilly

46

Ось функція, яку я написав для цього загального завдання. Це ефективніше, ніж Series/ stackметоди. Порядок і назви стовпців зберігаються.

def tidy_split(df, column, sep='|', keep=False):
    """
    Split the values of a column and expand so the new DataFrame has one split
    value per row. Filters rows where the column is missing.

    Params
    ------
    df : pandas.DataFrame
        dataframe with the column to split and expand
    column : str
        the column to split and expand
    sep : str
        the string used to split the column's values
    keep : bool
        whether to retain the presplit value as it's own row

    Returns
    -------
    pandas.DataFrame
        Returns a dataframe with the same columns as `df`.
    """
    indexes = list()
    new_values = list()
    df = df.dropna(subset=[column])
    for i, presplit in enumerate(df[column].astype(str)):
        values = presplit.split(sep)
        if keep and len(values) > 1:
            indexes.append(i)
            new_values.append(presplit)
        for value in values:
            indexes.append(i)
            new_values.append(value)
    new_df = df.iloc[indexes, :].copy()
    new_df[column] = new_values
    return new_df

З цією функцією вихідне питання настільки ж просто, як:

tidy_split(a, 'var1', sep=',')

1
Це блискуче швидко! Велике спасибі за це.
Анураг Н. Шарма

42

Панди> = 0,25

Методи Series та DataFrame визначають .explode()метод, який розриває списки в окремі рядки. Дивіться розділ "Документи" у розділі Вибух стовпця, що нагадує список .

Оскільки у вас є список рядків, розділених комами, розділіть рядок на коми, щоб отримати список елементів, а потім зателефонуйте explodeдо цього стовпця.

df = pd.DataFrame({'var1': ['a,b,c', 'd,e,f'], 'var2': [1, 2]})
df
    var1  var2
0  a,b,c     1
1  d,e,f     2

df.assign(var1=df['var1'].str.split(',')).explode('var1')

  var1  var2
0    a     1
0    b     1
0    c     1
1    d     2
1    e     2
1    f     2

Зауважте, що explodeпрацює лише в одному стовпчику (поки що).


NaNs та порожні списки отримують належне лікування, без того, щоб вам потрібно було стрибати через обручі, щоб виправити це.

df = pd.DataFrame({'var1': ['d,e,f', '', np.nan], 'var2': [1, 2, 3]})
df
    var1  var2
0  d,e,f     1
1            2
2    NaN     3

df['var1'].str.split(',')

0    [d, e, f]
1           []
2          NaN

df.assign(var1=df['var1'].str.split(',')).explode('var1')

  var1  var2
0    d     1
0    e     1
0    f     1
1          2  # empty list entry becomes empty string after exploding 
2  NaN     3  # NaN left un-touched

Це серйозна перевага перед рішеннями на базі ravel+repeat (які повністю ігнорують порожні списки та задушуються над NaN).


4
Цей найпростіший і найкраще в моєму випадку! Дякую!
Ісаак Сім

14

Подібне запитання, як: pandas: Як розділити текст у стовпці на кілька рядків?

Ви можете зробити:

>> a=pd.DataFrame({"var1":"a,b,c d,e,f".split(),"var2":[1,2]})
>> s = a.var1.str.split(",").apply(pd.Series, 1).stack()
>> s.index = s.index.droplevel(-1)
>> del a['var1']
>> a.join(s)
   var2 var1
0     1    a
0     1    b
0     1    c
1     2    d
1     2    e
1     2    f

2
Він працює після додавання ще одного коду перейменування s.name = 'var1'
Jesse

14

TL; DR

import pandas as pd
import numpy as np

def explode_str(df, col, sep):
    s = df[col]
    i = np.arange(len(s)).repeat(s.str.count(sep) + 1)
    return df.iloc[i].assign(**{col: sep.join(s).split(sep)})

def explode_list(df, col):
    s = df[col]
    i = np.arange(len(s)).repeat(s.str.len())
    return df.iloc[i].assign(**{col: np.concatenate(s)})

Демонстрація

explode_str(a, 'var1', ',')

  var1  var2
0    a     1
0    b     1
0    c     1
1    d     2
1    e     2
1    f     2

Створимо новий фрейм даних dіз списками

d = a.assign(var1=lambda d: d.var1.str.split(','))

explode_list(d, 'var1')

  var1  var2
0    a     1
0    b     1
0    c     1
1    d     2
1    e     2
1    f     2

Загальні коментарі

Я буду використовувати np.arangeз repeatвиробляти dataframe позиції індексу , який можна використовувати зiloc .

FAQ

Чому я не використовую loc ?

Тому що індекс може бути не унікальним і використовуючим loc поверне кожен рядок, який відповідає запитуваному індексу.

Чому ви не використовуєте цей valuesатрибут та фрагмент?

При виклику values, якщо вся фрейм даних знаходиться в одному згуртованому "блоці", Pandas поверне вигляд масиву, який є "блоком". Інакше Pandas доведеться обмотати новий масив. При обмотуванні цей масив повинен мати однаковий тип. Часто це означає повернення масиву з dtype, який є object. Використовуючи ilocзамість нарізкиvalues атрибут, я позбавляю себе від того, щоб з цим боротися.

Для чого ви використовуєте assign ?

Коли я користуюся assign те саме ім’я стовпця, що і я вибухаю, я перезаписую існуючий стовпець і підтримую його положення в кадрі даних.

Чому значення індексу повторюються?

Завдяки використанню ilocна повторних позиціях отриманий індекс показує таку ж повторювану схему. По одному повторення для кожного елемента списку або рядка.
Це можна скинути за допомогоюreset_index(drop=True)


Для струн

Мені не хочеться передчасно розбивати струни. Тож замість цього я підраховую випадки sepаргументу, припускаючи, що якби я розділив, довжина результуючого списку була б на одну більшу, ніж кількість роздільників.

Потім я використовую це sepдо joinрядків split.

def explode_str(df, col, sep):
    s = df[col]
    i = np.arange(len(s)).repeat(s.str.count(sep) + 1)
    return df.iloc[i].assign(**{col: sep.join(s).split(sep)})

Для списків

Подібно до рядків, за винятком того, що мені не потрібно рахувати подій sep оскільки його вже розщеплено.

Я використовую Numpy's, concatenateщоб загадати списки разом.

import pandas as pd
import numpy as np

def explode_list(df, col):
    s = df[col]
    i = np.arange(len(s)).repeat(s.str.len())
    return df.iloc[i].assign(**{col: np.concatenate(s)})


Мені це подобається. Справді лаконічна, і продуктивність теж повинна бути дуже хорошою. Хоча одне питання: чи df.iloc [i] те саме, що повторює рядки фрейму даних чи він більш ефективний? Дякую!
Тім

7

Існує можливість розділити та вибухнути фрейм даних без зміни структури фрейму даних

Розбийте та розгорніть дані конкретних стовпців

Вхід:

    var1    var2
0   a,b,c   1
1   d,e,f   2



#Get the indexes which are repetative with the split 
temp = df['var1'].str.split(',')
df = df.reindex(df.index.repeat(temp.apply(len)))


df['var1'] = np.hstack(temp)

Вийшов:

    var1    var2
0   a   1
0   b   1
0   c   1
1   d   2
1   e   2
1   f   2

Правка-1

Розділити та розгорнути рядки для кількох стовпців

Filename    RGB                                             RGB_type
0   A   [[0, 1650, 6, 39], [0, 1691, 1, 59], [50, 1402...   [r, g, b]
1   B   [[0, 1423, 16, 38], [0, 1445, 16, 46], [0, 141...   [r, g, b]

Повторна індексація на основі стовпчика посилань та вирівнювання інформації про значення стовпця зі стеком

df = df.reindex(df.index.repeat(df['RGB_type'].apply(len)))
df = df.groupby('Filename').apply(lambda x:x.apply(lambda y: pd.Series(y.iloc[0])))
df.reset_index(drop=True).ffill()

Вийшов:

                Filename    RGB_type    Top 1 colour    Top 1 frequency Top 2 colour    Top 2 frequency
    Filename                            
 A  0       A   r   0   1650    6   39
    1       A   g   0   1691    1   59
    2       A   b   50  1402    49  187
 B  0       B   r   0   1423    16  38
    1       B   g   0   1445    16  46
    2       B   b   0   1419    16  39

5

Я придумав рішення для фреймів даних з довільною кількістю стовпців (при цьому все одно розділяючи лише записи одного стовпчика одночасно).

def splitDataFrameList(df,target_column,separator):
    ''' df = dataframe to split,
    target_column = the column containing the values to split
    separator = the symbol used to perform the split

    returns: a dataframe with each entry for the target column separated, with each element moved into a new row. 
    The values in the other columns are duplicated across the newly divided rows.
    '''
    def splitListToRows(row,row_accumulator,target_column,separator):
        split_row = row[target_column].split(separator)
        for s in split_row:
            new_row = row.to_dict()
            new_row[target_column] = s
            row_accumulator.append(new_row)
    new_rows = []
    df.apply(splitListToRows,axis=1,args = (new_rows,target_column,separator))
    new_df = pandas.DataFrame(new_rows)
    return new_df

2
приємно, але на жаль повільно через цю конверсію todict () :(
MAQ

4

Ось досить відверте повідомлення, яке використовує splitметод з пандstr аксесуара а потім використовує NumPy для вирівнювання кожного рядка в один масив.

Відповідні значення отримують шляхом повторення стовпця, що не розщеплюється, правильну кількість разів np.repeat.

var1 = df.var1.str.split(',', expand=True).values.ravel()
var2 = np.repeat(df.var2.values, len(var1) / len(df))

pd.DataFrame({'var1': var1,
              'var2': var2})

  var1  var2
0    a     1
1    b     1
2    c     1
3    d     2
4    e     2
5    f     2

1
Це може бути дуже красивою відповіддю. На жаль, він не масштабує для багатьох стовпців, чи не так?
Майкл Дорнер

3

Я боровся зі споживанням пам'яті, використовуючи різні способи вибуху моїх списків, тому я підготував декілька орієнтирів, які допоможуть мені вирішити, на які відповіді звертатись. Я перевірив п'ять сценаріїв із різною пропорцією довжини списку до кількості списків. Надання результатів нижче:

Час: (менше краще, натисніть, щоб переглянути велику версію)

Швидкість

Пік використання пам'яті: (менше краще)

Пік використання пам'яті

Висновки :

  • @ Відповідь MaxU (оновлення 2), об'єднане кодове ім'я пропонує найкращу швидкість майже в кожному випадку, зберігаючи при цьому низький рівень використання пам'яті,
  • див. відповідь @ DMulligan ( стек кодових імен ), якщо вам потрібно обробити багато рядків із відносно невеликими списками та можете дозволити собі збільшити пік пам'яті,
  • відповідь прийнятого @ Чанга добре працює для кадрів даних, які мають кілька рядків, але дуже великі списки.

Повна інформація (функції та код бенчмаркінгу) міститься в цій суті GitHub . Зверніть увагу, що проблема еталону була спрощена і не включала розбиття рядків у список - які більшість рішень виконували аналогічно.


Приємне порівняння! Ви не хочете розмістити код, який ви використовували для побудови орієнтирів?
MaxU

1
Перейдіть за цим посиланням: gist.github.com/krassowski/0259a2cd2ba774ccd9f69bbcc3187fbf (вже включено у відповідь) - IMO було б трохи задовго, щоб все це вставити.
krassowski

2

Грунтуючись на відмінно @ DMulligan в розчині , тут векторизация функція загальна (без петель) , який розбиває стовпець dataframe на кілька рядків і поміщає її назад до первісного dataframe. З change_column_orderцієї відповіді він також використовує чудову загальну функцію .

def change_column_order(df, col_name, index):
    cols = df.columns.tolist()
    cols.remove(col_name)
    cols.insert(index, col_name)
    return df[cols]

def split_df(dataframe, col_name, sep):
    orig_col_index = dataframe.columns.tolist().index(col_name)
    orig_index_name = dataframe.index.name
    orig_columns = dataframe.columns
    dataframe = dataframe.reset_index()  # we need a natural 0-based index for proper merge
    index_col_name = (set(dataframe.columns) - set(orig_columns)).pop()
    df_split = pd.DataFrame(
        pd.DataFrame(dataframe[col_name].str.split(sep).tolist())
        .stack().reset_index(level=1, drop=1), columns=[col_name])
    df = dataframe.drop(col_name, axis=1)
    df = pd.merge(df, df_split, left_index=True, right_index=True, how='inner')
    df = df.set_index(index_col_name)
    df.index.name = orig_index_name
    # merge adds the column to the last place, so we need to move it back
    return change_column_order(df, col_name, orig_col_index)

Приклад:

df = pd.DataFrame([['a:b', 1, 4], ['c:d', 2, 5], ['e:f:g:h', 3, 6]], 
                  columns=['Name', 'A', 'B'], index=[10, 12, 13])
df
        Name    A   B
    10   a:b     1   4
    12   c:d     2   5
    13   e:f:g:h 3   6

split_df(df, 'Name', ':')
    Name    A   B
10   a       1   4
10   b       1   4
12   c       2   5
12   d       2   5
13   e       3   6
13   f       3   6    
13   g       3   6    
13   h       3   6    

Зауважте, що він зберігає початковий індекс та порядок стовпців. Він також працює з фреймами даних, які мають не послідовний індекс.


2
це зламало це для мене, приємна робота: stackoverflow.com/a/48554655/6672746
Еван

2

Функція розбиття рядка може приймати опцію булевого аргументу "розширити".

Ось рішення з використанням цього аргументу:

(a.var1
  .str.split(",",expand=True)
  .set_index(a.var2)
  .stack()
  .reset_index(level=1, drop=True)
  .reset_index()
  .rename(columns={0:"var1"}))

1

Щойно використаний відмінний відповідь jiln зверху, але його потрібно розширити, щоб розділити кілька стовпців. Думав, що поділюсь.

def splitDataFrameList(df,target_column,separator):
''' df = dataframe to split,
target_column = the column containing the values to split
separator = the symbol used to perform the split

returns: a dataframe with each entry for the target column separated, with each element moved into a new row. 
The values in the other columns are duplicated across the newly divided rows.
'''
def splitListToRows(row, row_accumulator, target_columns, separator):
    split_rows = []
    for target_column in target_columns:
        split_rows.append(row[target_column].split(separator))
    # Seperate for multiple columns
    for i in range(len(split_rows[0])):
        new_row = row.to_dict()
        for j in range(len(split_rows)):
            new_row[target_columns[j]] = split_rows[j][i]
        row_accumulator.append(new_row)
new_rows = []
df.apply(splitListToRows,axis=1,args = (new_rows,target_column,separator))
new_df = pd.DataFrame(new_rows)
return new_df

1

оновив відповідь MaxU за допомогою підтримки MultiIndex

def explode(df, lst_cols, fill_value='', preserve_index=False):
    """
    usage:
        In [134]: df
        Out[134]:
           aaa  myid        num          text
        0   10     1  [1, 2, 3]  [aa, bb, cc]
        1   11     2         []            []
        2   12     3     [1, 2]      [cc, dd]
        3   13     4         []            []

        In [135]: explode(df, ['num','text'], fill_value='')
        Out[135]:
           aaa  myid num text
        0   10     1   1   aa
        1   10     1   2   bb
        2   10     1   3   cc
        3   11     2
        4   12     3   1   cc
        5   12     3   2   dd
        6   13     4
    """
    # make sure `lst_cols` is list-alike
    if (lst_cols is not None
        and len(lst_cols) > 0
        and not isinstance(lst_cols, (list, tuple, np.ndarray, pd.Series))):
        lst_cols = [lst_cols]
    # all columns except `lst_cols`
    idx_cols = df.columns.difference(lst_cols)
    # calculate lengths of lists
    lens = df[lst_cols[0]].str.len()
    # preserve original index values    
    idx = np.repeat(df.index.values, lens)
    res = (pd.DataFrame({
                col:np.repeat(df[col].values, lens)
                for col in idx_cols},
                index=idx)
             .assign(**{col:np.concatenate(df.loc[lens>0, col].values)
                            for col in lst_cols}))
    # append those rows that have empty lists
    if (lens == 0).any():
        # at least one list in cells is empty
        res = (res.append(df.loc[lens==0, idx_cols], sort=False)
                  .fillna(fill_value))
    # revert the original index order
    res = res.sort_index()
    # reset index if requested
    if not preserve_index:        
        res = res.reset_index(drop=True)

    # if original index is MultiIndex build the dataframe from the multiindex
    # create "exploded" DF
    if isinstance(df.index, pd.MultiIndex):
        res = res.reindex(
            index=pd.MultiIndex.from_tuples(
                res.index,
                names=['number', 'color']
            )
    )
    return res

1

Один вкладиш з використанням split(___, expand=True)і levelта nameаргументи reset_index():

>>> b = a.var1.str.split(',', expand=True).set_index(a.var2).stack().reset_index(level=0, name='var1')
>>> b
   var2 var1
0     1    a
1     1    b
2     1    c
0     2    d
1     2    e
2     2    f

Якщо вам потрібно bвиглядати точно так, як у запитанні, ви можете додатково зробити:

>>> b = b.reset_index(drop=True)[['var1', 'var2']]
>>> b
  var1  var2
0    a     1
1    b     1
2    c     1
3    d     2
4    e     2
5    f     2

0

Я придумав таке рішення цієї проблеми:

def iter_var1(d):
    for _, row in d.iterrows():
        for v in row["var1"].split(","):
            yield (v, row["var2"])

new_a = DataFrame.from_records([i for i in iter_var1(a)],
        columns=["var1", "var2"])

0

Ще одне рішення, яке використовує пакет копіювання python

import copy
new_observations = list()
def pandas_explode(df, column_to_explode):
    new_observations = list()
    for row in df.to_dict(orient='records'):
        explode_values = row[column_to_explode]
        del row[column_to_explode]
        if type(explode_values) is list or type(explode_values) is tuple:
            for explode_value in explode_values:
                new_observation = copy.deepcopy(row)
                new_observation[column_to_explode] = explode_value
                new_observations.append(new_observation) 
        else:
            new_observation = copy.deepcopy(row)
            new_observation[column_to_explode] = explode_values
            new_observations.append(new_observation) 
    return_df = pd.DataFrame(new_observations)
    return return_df

df = pandas_explode(df, column_name)

0

Тут є багато відповідей, але я здивований, що ніхто не згадав про вбудовану функцію панди. Перевірте посилання нижче: https://pandas.pydata.org/pandas-docs/stable/reference/api/pandas.DataFrame.explode.html#pandas.DataFrame.explode

Чомусь мені не вдалося отримати доступ до цієї функції, тому я використав код нижче:

import pandas_explode
pandas_explode.patch()
df_zlp_people_cnt3 = df_zlp_people_cnt2.explode('people')

введіть тут опис зображення

Вище - зразок моїх даних. Як ви бачите, у колонці людей було ряд людей, і я намагався її вибухнути. Код, який я надав, працює для даних типу списку. Тому спробуйте перевести текстові дані, розділені комами, у формат списку. Оскільки мій код використовує вбудовані функції, він набагато швидший, ніж спеціальні / застосовні функції.

Примітка. Можливо, вам доведеться встановити pandas_explode з pip.


0

У мене була подібна проблема, моє рішення - перетворити кадр даних спочатку в список словників, а потім зробити перехід. Ось функція:

import copy
import re

def separate_row(df, column_name):
    ls = []
    for row_dict in df.to_dict('records'):
        for word in re.split(',', row_dict[column_name]):
            row = copy.deepcopy(row_dict)
            row[column_name]=word
            ls(row)
    return pd.DataFrame(ls)

Приклад:

>>> from pandas import DataFrame
>>> import numpy as np
>>> a = DataFrame([{'var1': 'a,b,c', 'var2': 1},
               {'var1': 'd,e,f', 'var2': 2}])
>>> a
    var1  var2
0  a,b,c     1
1  d,e,f     2
>>> separate_row(a, "var1")
  var1  var2
0    a     1
1    b     1
2    c     1
3    d     2
4    e     2
5    f     2

Ви також можете трохи змінити функцію для підтримки роздільних рядків типу списку.

Використовуючи наш веб-сайт, ви визнаєте, що прочитали та зрозуміли наші Політику щодо файлів cookie та Політику конфіденційності.
Licensed under cc by-sa 3.0 with attribution required.