The most advanced ML Observability product in the market
Building an ML platform is nothing like putting together Ikea furniture; obviously, Ikea is way more difficult. However, they both, similarly, include many different parts that help create value when put together. As every organization sets out on a unique path to building its own machine learning platform, taking on the project of building a […]
Start integrating our products and tools.
We’re excited 😁 to share that Forbes has named Aporia a Next Billion-Dollar Company. This recognition comes on the heels of our recent $25 million Series A funding and is a huge testament that Aporia’s mission and the need for trust in AI are more relevant than ever. We are very proud to be listed […]
The DataFrames we work with in real life are quite large and contain lots of columns. In some cases, it is not practical to visually check the column names and we want them in a list.
In this short how-to article, we will learn how to create a list from column names in Pandas and PySpark DataFrames.
The columns method returns an Index object which contains all the column names. It can be converted to a list by using the list constructor or the tolist method.
# with list constructor
col_list = list(df.columns)
# with tolist method
col_list = df.columns.tolist()
The columns method in PySpark returns a list of columns. Thus, we do not need an additional operation.
col_list = df.columns