0 1 2 3 4 5 6 7 8 9
3 False True False True True False True True True False
4 False True False True True True True True False False
I need to create something like this
dict= {3: [1,3,4,6,7,8], 4: [0,1,3,4,5,6,7]}
CodePudding user response:
I hope I've understood your question right:
out = dict(
df.apply(lambda x: [int(i) for i, v in zip(x.index, x) if v], axis=1)
)
print(out)
Prints:
{3: [1, 3, 4, 6, 7, 8], 4: [1, 3, 4, 5, 6, 7]}