Create a list of namedtuples from a dataframe

Question

I have a dataframe like this:

df1

Name   Category  Age
Harry   A        11
James   B        23
Will    A        19

I want to create a list of tuples using namedtuple from collections. The list should be like this:

output_list = [Variable(Name='Harry', Age=11), Variable(Name='James', Age=23), Variable(Name='Will', Age=19)]

This is what I've tried using 'itertuples'

output_list = list(df1[["Name","Age"]].itertuples(name='Variable', index=False))

In fact, I don't understand what is your question. The second form is what you want, no? And it's probably better than my answer using `namedtuple` — Corralien, Jun 01 '22 at 20:01

Corralien · Accepted Answer · 2022-06-01T20:00:52.770

1

Try:

from collections import namedtuple

COLS = ['Name', 'Age']
Variable = namedtuple('Variable', field_names=COLS)
output_list = df[COLS].apply(lambda x: Variable(**x), axis=1).tolist()
print(output_list)

# Output
[Variable(Name='Harry', Age=11),
 Variable(Name='James', Age=23),
 Variable(Name='Will', Age=19)]

edited Jun 01 '22 at 20:00

answered Jun 01 '22 at 19:53

Corralien

109,409
8
28
52

He doesn't have the "Category" in his output example :) – Drakax Jun 01 '22 at 19:54
Why is this better than the OP's solution? – juanpa.arrivillaga Jun 01 '22 at 20:03
@juanpa.arrivillaga. Did you read my comment above? – Corralien Jun 01 '22 at 20:04
yes? Again, why is doing this better than `list(df1[["Name","Age"]].itertuples(name='Variable', index=False))`? – juanpa.arrivillaga Jun 01 '22 at 20:04
From my comment to the question: *And it's probably better than my answer using namedtuple* – Corralien Jun 01 '22 at 20:05
Oh, you meant in the answer above. Yeah ok, you seemed to have noticed as well. – juanpa.arrivillaga Jun 01 '22 at 20:05
Look, it's probably best to ask for clarification from the OP, honestly, the question should be closed because it isn't clear *what is being asked*. – juanpa.arrivillaga Jun 01 '22 at 20:06
Maybe you are right but the OP asks *I want to create a list of tuples using namedtuple from collections*. It's what my answer does :-P – Corralien Jun 01 '22 at 20:07

score 0 · Answer 2 · answered Jun 01 '22 at 19:53

0

Maybe not the answer you're looking for but:

tuples = [tuple(x) for x in df1[['Name','Age']].to_numpy()]
tuples

Output:

[('Harry', 11), ('James', 23), ('Will', 19)]

answered Jun 01 '22 at 19:53

Drakax

1,305
3
9

Why is this better than the OP's stated solution? – juanpa.arrivillaga Jun 01 '22 at 20:03

Create a list of namedtuples from a dataframe

2 Answers2