pyspark 拿到列名list
时间: 2023-09-14 10:11:22 浏览: 103
mybatis 批量将list数据插入到数据库的实现
可以使用DataFrame的columns属性来获取列名列表,示例如下:
```python
from pyspark.sql import SparkSession
spark = SparkSession.builder.appName("get_column_names").getOrCreate()
df = spark.read.csv("path/to/csv/file", header=True)
column_names = df.columns
print(column_names)
```
输出结果为:
```
['col1', 'col2', 'col3']
```
阅读全文