Find names of top-n highest-value columns in each pandas dataframe row

by

You could use np.argsort to find the indices of the n largest items for each row:

import numpy as np
import pandas as pd

df = pd.DataFrame({'id': [1, 2, 3, 4, 5],
 'p1': [0, 0, 1, 1, 2],
 'p2': [9, 2, 3, 5, 3],
 'p3': [1, 3, 10, 3, 7],
 'p4': [4, 4, 7, 1, 10]})
df = df.set_index('id')

nlargest = 3
order = np.argsort(-df.values, axis=1)[:, :nlargest]
result = pd.DataFrame(df.columns[order], 
                      columns=['top{}'.format(i) for i in range(1, nlargest+1)],
                      index=df.index)

print(result)

yields

   top1 top2 top3
id               
1    p2   p4   p3
2    p4   p3   p2
3    p3   p4   p2
4    p2   p3   p1
5    p4   p3   p2

More Related Contents:

  1. how to calculation cost time [closed]
  2. Filter dataframe rows if value in column is in a set list of values [duplicate]
  3. Converting a Pandas GroupBy output from Series to DataFrame
  4. String concatenation of two pandas columns
  5. pandas: filter rows of DataFrame with operator chaining
  6. datetime dtypes in pandas read_csv
  7. Reshape wide to long in pandas
  8. How do I create test and train samples from one dataframe with pandas?
  9. Pandas – Slice large dataframe into chunks
  10. Calculate average of every x rows in a table and create new table
  11. Find column name in pandas that matches an array
  12. How do convert a pandas dataframe to XML?
  13. Sort a pandas dataframe series by month name
  14. Count frequency of values in pandas DataFrame column
  15. Strip / trim all strings of a dataframe
  16. Joining pandas DataFrames by Column names
  17. pandas get the row-wise minimum value of two or more columns
  18. Set maximum value (upper bound) in pandas DataFrame
  19. Diff of two Dataframes
  20. Normalizing pandas DataFrame rows by their sums
  21. Apply function to pandas DataFrame that can return multiple rows
  22. How can I get a value from a cell of a dataframe?
  23. How to add header row to a pandas DataFrame
  24. creating a column which keeps a running count of consecutive values
  25. Select row by max value in group in a pandas dataframe
  26. How to drop column according to NAN percentage for dataframe?
  27. Extracting dates that are in different formats using regex and sorting them – pandas
  28. Sum columns by level in a pandas MultiIndex DataFrame
  29. pandas extract year from datetime: df[‘year’] = df[‘date’].year is not working
  30. How to upload a CSV file in FastAPI and convert it into a Pandas Dataframe?

Leave a Comment