[scikit-learn] Identifying column names of Non-zero values

[Startup Hire]

Hi Samo,

Thanks a lot. It works at a row level and I can append it a row level to
the main dataframe to do further analysis.

Regards,
Sanant

On Wed, Oct 5, 2016 at 5:05 PM, Samo Turk <samo.turk at gmail.com> wrote:

> Something like this might work:
>
> def non_zero(row, columns):
>     return list(columns[~(row == 0)])
>
> df.apply(lambda x: non_zero(x, df.columns), axis=1)
>
> Cheers,
> Samo
>
> On Wed, Oct 5, 2016 at 11:58 AM, Startup Hire <blrstartuphire at gmail.com>
> wrote:
>
>> Hi Pypers,
>>
>> Hope you are doing well.
>>
>> I am working on a project to find out the column names of non-zero values
>> at a row level.
>>
>> How can this effectively done in python pandas/dataframe?
>>
>>
>> For example,
>>
>>
>>
>>
>>
>>
>>
>>
>>
>>
>>
>>
>>
>>
>>
>> *Column1* *Column *2 *Column *3 Column 4 Column 5 Column 6 *Column 7* New
>> column to be created
>> 1 1 1 0 0 0 1 Column1,Column 2,Column 3,Column7
>>
>>
>>
>>
>>
>>
>>
>>
>>
>>
>> I might have to do it on approximately million rows
>>
>> Regards,
>> Sanant
>>
>> _______________________________________________
>> scikit-learn mailing list
>> scikit-learn at python.org
>> https://mail.python.org/mailman/listinfo/scikit-learn
>>
>>
>
> _______________________________________________
> scikit-learn mailing list
> scikit-learn at python.org
> https://mail.python.org/mailman/listinfo/scikit-learn
>
>
-------------- next part --------------
An HTML attachment was scrubbed...
URL: <http://mail.python.org/pipermail/scikit-learn/attachments/20161005/f3ca680c/attachment.html>