擅长:python、mysql、java
<p>你能用这样的<code>numpy.argmax</code>吗:</p>
<pre><code>import numpy as np
# This is a simple example. In your case, A is loaded with np.loadtxt
A = np.array([[1, 2.0, 3.0], [3, 1.0, 2.0], [2.0, 4.0, 3.0]])
B = A.copy()
# Copy the max indices of rows of A into first column of B
B[:,0] = np.argmax(A[:,1:], 1)
# Save the results using np.savetxt with fmt, dynamically generating the
# format string based on the number of columns in B (setting the first
# column to integer and the rest to float)
np.savetxt('/path/to/output.txt', B, fmt='%d' + ' %f' * (B.shape[1]-1))
</code></pre>
<p>注意,<code>np.savetxt</code>允许格式化。你知道吗</p>
<p>此示例代码没有说明您希望跳过第一行的事实,您可能希望从<code>np.argmax</code>的结果中减去1,这取决于剩余列中的索引是否包含索引列(0)。你知道吗</p>