如果列不存在,我不想报错
时间: 2023-08-19 19:27:10 浏览: 46
如果你想要根据列名选取列,但是又不确定该列是否存在,可以使用`DataFrame`对象的`get()`方法。如果该列存在,则返回该列的数据;否则,返回指定的默认值(默认为`None`)。
例如,假设有一个名为`df`的`DataFrame`对象,其中可能包含名为`col1`的列。可以使用以下代码来安全地选取该列:
```python
col1_data = df.get('col1')
```
如果`col1`列存在,则返回该列的数据;否则,返回`None`。
注意,`get()`方法只能根据列名选取列,不能根据列索引选取列。如果你想要根据列索引选取列,可以使用`iloc`方法或`[]`操作符。如果列不存在,则会抛出`KeyError`异常。如果你不想报错,可以使用`try-except`语句捕获异常,或者使用`DataFrame`对象的`columns`属性获取所有列名,然后检查指定的列名是否存在。