Convert dataframe of list in columns to rows

Andrés Bustamante :

I have a pandas DataFrame of this type

col1  col2           col3
 1   [blue]         [in,out]
 2   [green, green] [in]
 3   [green]        [in]

and I need convert it to a dataframe that keep the first column and distribute all the other values in columns as rows:

col1 value
1    blue
1    in
1    out
2    green
2    green
2    in
3    green
3    in

jezrael :

Use DataFrame.stack with Series.explode for convert lists, last some data cleaning with DataFrame.reset_index:

df1 = (df.set_index('col1')
         .stack()
         .explode()
         .reset_index(level=1, drop=True)
         .reset_index(name='value'))

Alternative with DataFrame.melt and DataFrame.explode:

df1 = (df.melt('col1')
         .explode('value')
         .sort_values('col1')[['col1','value']]
         .reset_index(drop=True)
)

---

print (df1)
   col1  value
0     1   blue
1     1     in
2     1    out
3     2  green
4     2  green
5     2     in
6     3  green
7     3     in

Or list comprehension solution:

L = [(k, x) for k, v in df.set_index('col1').to_dict('index').items() 
            for k1, v1 in v.items() 
            for x in v1]

df1 = pd.DataFrame(L, columns=['col1','value'])
print (df1)
   col1  value
0     1   blue
1     1     in
2     1    out
3     2  green
4     2  green
5     2     in
6     3  green
7     3     in

Guess you like

Origin http://43.154.161.224:23101/article/api/json?id=279151&siteId=1

Convert dataframe of list in columns to rows

pandas: get rows by comparing two columns of dataframe to list of tuples

pandas gets the number of rows, columns, and elements of the dataframe

pandas select rows by condition for all of dataframe columns

DataFrame deletes all rows or columns with a certain value

Convert many column into multiple columns and rows in mysql

Enrich data in rows and columns in ofbiz list component

python dataframe columns and rows to get the column names the name index

how to remove rows that appear same in two columns simultaneously in dataframe?

Selecting dataframe rows based on multiple columns, where new functions should be created to handle conditions in some columns

Python how to combine two columns of a dataframe into a single list?

Transform pandas dataframe columns to list according to number in row

Pandas DataFrame how to group (pivot?) rows by values of specified columns, but keeping the original index?

The text consists of rows into columns

Table columns and rows Merged

Rows and columns in mysql sql

tableWidget hide rows and columns

SQL - query columns into rows

Layout across rows or columns

Reshape dataframe using pandas/numpy in a particular manner - convert multiple columns into two

Use Python to convert the data types of all columns in dataframe format to categorical data types

choosing rows by values in DataFrame

Moving rows in multiple columns into their own columns Python

Moving rows in multiple columns into their own columns Python

Order columns in pandas dataframe

js- initial array is a two-dimensional array of 4 rows 3. Convert it to a two-dimensional array of four rows and three columns

MySQL rows into columns and columns rows column transfer switch

antd table merge rows or columns (dynamically add merged rows and columns)

The grid system container rows and columns

MapReduce removing duplicate rows, columns