Fast Way to Get Index of Top-K Elements of Every Column in a Pandas DataFrame

Understanding Integer Indexing in Pandas Series and DataFrames: A Guide to Label-Based and Integer-Based Indexing.