擅长:python、mysql、java
<p>与使用<code>np.transpose</code>(这是1D数据的no-op)不同,您只需在将维度转换为数组后添加一个维度:</p>
<pre><code>tr = np.array(col_1)[:, None] # instead of "tr = np.transpose(col_1)"
</code></pre>
<p>例如:</p>
^{pr2}$
<p>还有其他几种改进代码的方法,但是这种更改应该可以解决问题。:)</p>
<p>例如,<a href="https://docs.scipy.org/doc/numpy/reference/generated/numpy.loadtxt.html#numpy-loadtxt" rel="nofollow noreferrer">^{<cd2>}</a>可用于以更直接的方式提取列:</p>
<pre><code># to extract the fifth column of a csv file:
tr = np.loadtxt(filename, delimiter=col_sep, comments='#')[:, 4][:, None]
</code></pre>