这个字典,前三列识别为字符串,后200列识别为浮点数,怎么写简便
时间: 2024-02-25 20:55:09 浏览: 70
可以使用列表推导式和字典推导式来生成这个字典,例如:
```python
dtypes = {col: str if i < 3 else float for i, col in enumerate(df.columns)}
```
这里的 `col` 表示列名,`df.columns` 表示数据框的所有列名。`enumerate()` 函数将列名和索引组合成一个元组,然后使用条件表达式判断当前列是否为前三列。如果是,将数据类型指定为字符串;否则,将数据类型指定为浮点数。最终将所有列名和对应的数据类型组成一个字典。
需要注意的是,这种写法假设所有列名都是字符串类型。如果你的数据框中存在其他类型的列名,需要进行额外的类型判断和转换。
阅读全文