Python 3中的紧凑切片符号

问题描述 投票:0回答:1

是否有使用Python 3编写以下代码的紧凑版本:

print(columns[1],columns[8],columns[9])

类似:

print(columns [1,8,9])

要获得相同的结果。

python-3.x slice
1个回答
0
投票

也许这吗?

import numpy as np
import pandas as pd
df = pd.DataFrame(np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]]),
                   columns=['a', 'b', 'c'])
print(df[['a','b']])
© www.soinside.com 2019 - 2024. All rights reserved.