Question

我这个python代码，我想知道split（）后[0]的用途是什么这是转置和展平代码

array = numpy.array([input().strip().split() for _ in range(int(input().split()[0]))], int)
print (array.transpose())
print (array.flatten()

结果：

 3
1 2 3
4 5 6
7 8 9
[[1 4 7]
 [2 5 8]
 [3 6 9]]
[1 2 3 4 5 6 7 8 9]

Answer 1

[0]返回列表的第一个元素。例如：

将a设置为字符串：

>>> a = '4 5 6'
>>> a
'4 5 6'

.split()返回（字符串列表）：

>>> a.split()
['4', '5', '6']

[0]返回列表的第一个元素。

>>> a.split()[0]
'4'

如果我们将输入字符串减少为一个数字：

>>> b = '3'
>>> b
'3'

.split()返回长度为1的列表

>>> b.split()
['3']

列表不能直接转换为整数：

>>> int(b.split())
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
TypeError: int() argument must be a string, a bytes-like object or a number, not 'list'

[0]返回列表的第一个元素，可以将其转换为一个int：

>>> b.split()[0]
'3'
>>> int(b.split()[0])
3

如果您知道字符串中只有一个整数，则不需要.split()[0]：

>>> int(b)
3

但是如果字符串包含多个数字，这将不起作用：

>>> int(a)
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
ValueError: invalid literal for int() with base 10: '4 5 6'
>>>

“ input（）。split（）[0]”中的“ [0]”的用途是什么

1 个答案: