是否有使用Python 3编写以下代码的紧凑版本:
print(columns[1],columns[8],columns[9])
类似:
print(columns [1,8,9])
要获得相同的结果。
也许这吗?
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']])