python - 从数据框中随机选择列

Question

我的问题很简单: 有没有办法从 Pandas 的数据框中随机选择列？明确地说，我想随机挑选出带有附加值的 n 列。我知道有这样一种随机选择行的方法:

import pandas as pd

df = pd.read_csv(filename, sep=',', nrows=None)
a = df.sample(n = 2)

所以问题是，是否存在用于查找随机列的等效方法？

Answer 1

sample 也接受一个轴参数:

df = pd.DataFrame(np.random.randint(1, 10, (10, 5)), columns=list('abcde'))

df
Out: 
   a  b  c  d  e
0  4  5  9  8  3
1  7  2  2  8  7
2  1  5  7  9  2
3  3  3  5  2  4
4  8  4  9  8  6
5  6  5  7  3  4
6  6  3  6  4  4
7  9  4  7  7  3
8  4  4  8  7  6
9  5  6  7  6  9

df.sample(2, axis=1)
Out: 
   a  d
0  4  8
1  7  8
2  1  9
3  3  2
4  8  8
5  6  3
6  6  4
7  9  7
8  4  7
9  5  6