How to create variable list of list of tuples from selected columns in dataframe? [duplicate]

Question

I have df that looks like this:

col1     col2      col3     col4
1         0         1        0 
1         0         1        0 
1         0         1        0 
1         0         1        0 
1         0         1        0

How do I convert col1-col3 into a list of list of tuples and assign to variable? and then create an array of only col4 and assign to variable?

example final result:

lst = [[1,0,1],[1,0,1],[1,0,1],[1,0,1],[1,0,1]]]
lst_col4 = [[0],[0],[0],[0],[0]]

You can select specific columns and then convert to a list of lists. See the marked duplicates. — jpp
– jpp, Commented Dec 18, 2018 at 17:16

Tim · Accepted Answer · 2018-12-18 17:18:32Z

4

Just index those coluns, and use the .values attribute:

lst = df[[col1, col2, col3]].values.tolist()

Example:

>>> df = pd.DataFrame({'a':[1,2,3], 'b':[4,5,6], 'c':[7,8,9]})
>>> lst = df[['a', 'c']].values.tolist()
>>> print(lst)
 [[1, 7], [2, 8],[3, 9]]

edited Dec 18, 2018 at 17:18

answered Dec 18, 2018 at 17:13

Tim

2,8531 gold badge18 silver badges33 bronze badges

Sign up to request clarification or add additional context in comments.

1 Comment

RustyShackleford Over a year ago

this did it thank you!

Collectives™ on Stack Overflow

How to create variable list of list of tuples from selected columns in dataframe? [duplicate]

1 Answer 1

1 Comment

Linked

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

1 Comment

Linked

Related