擅长:python、mysql、java
<p>您可以创建一个返回行和列元组的函数,如下所示:</p>
<pre><code>def rowsCols(a):
if len(a.shape) > 1:
rows = a.shape[0]
cols = a.shape[1]
else:
rows = a.shape[0]
cols = 0
return (rows, cols)
</code></pre>
<p>其中<code>a</code>是您输入到函数的数组。下面是使用函数的示例:</p>
^{pr2}$
<p>这将打印<code>rows 3 and columns 0</code>,然后<code>rows 4 and columns 3</code>。或者,您可以使用<code>atleast_2d</code>函数来实现更简洁的方法:</p>
<pre><code>(r, c) = np.atleast_2d(x).shape
print('rows {}, cols {}'.format(r, c))
(r, c) = np.atleast_2d(y).shape
print('rows {}, cols {}'.format(r, c))
</code></pre>
<p>它打印<code>rows 1, cols 3</code>和{<cd6>}。在</p>