python - 识别最高行值所属的列，python或R

Question

所以我要创建一个由 x 行组成的 20 列的表格，我需要为每一行找到最高值属于哪一列。例如:

The Table would be something like this (but larger)
 A       B       C       D       E       F       G
 1       2       3       4       5       6       7
 9       8       7       6       5       4       3
 7       6       5       8       4       3       2
 0.9     0.01    0.02    0.2     0.04    0.3   ...

我希望它吐出:G,A,D,A.我需要把它放到另一个文件中。它甚至不必与字母一起出现。稍后我会用它做点什么。我一直在努力找出最好的方法，并且一直在研究尝试用 R 来做，这是我到目前为止的脚本:

#!/usr/bin/env Rscript
a=read.table(get(TEST.csv),header=T,sep="",dec=".")
apply(a, 1, which.max)

它不想读取我的测试文件。对于 python，我有以下内容:

import numpy as np
import csv
a=np.genfromtxt('./TEST.csv',delimiter='\t',skip_header=1)
print(a)
amax=np.amax(a,axis=1)
print(amax)

这个正确地提取了每一行的最高值，但它没有像我希望的那样提取列号。任何和所有建议将不胜感激。

Answer 1

你可以在 R 中尝试 max.col

names(a)[max.col(a, 'first')]
#[1] "G" "A" "D" "A"