python - pandas DataFrame.drop 函数中的整数参数

Question

在以下代码行中

 X = np.array(df.drop(['label'], 1))

您能解释一下数字 1 的作用吗？

从文档中我了解到 DataFrame.drop 函数从数据帧中删除名为 'label' 的所需列，并返回没有此列的新数据帧。但我不明白这个特定的整数参数 1 是做什么的。

Answer 1

是drop中的参数axis 。它与axis=1相同。这意味着您需要从第一个参数labels中指定的DataFrame中删除列:

labels 大多数情况下都会被省略。
如果需要删除带有index的行，可以删除参数axis，因为默认情况下axis=0。参数axis=1有时会被替换为1，因为文字较少，但可读性较差。

示例:

import pandas as pd

df = pd.DataFrame({'label':[1,2,3],
                   'label1':[4,5,6],
                   'label2':[7,8,9]})

print (df)
   label  label1  label2
0      1       4       7
1      2       5       8
2      3       6       9

print (df.drop(['label'], 1))
   label1  label2
0       4       7
1       5       8
2       6       9

#most commonly used
print (df.drop(['label'], axis=1))
   label1  label2
0       4       7
1       5       8
2       6       9

print (df.drop(labels=['label'], axis=1))
   label1  label2
0       4       7
1       5       8
2       6       9