Python: Pandas in a nutshell

• Python
• Self-Learning

Note: In the following of this post, the responses are those of ChatGPT after being aware of a basic script.

Retrieving DataFrame Headers

Yes, it is entirely possible to print only the headers of a DataFrame with pandas. Here’s how you can do it:

import pandas as pd

data = pd.DataFrame({
    'Column1': [1, 2, 3],
    'Column2': [4, 5, 6],
    'Column3': [7, 8, 9]
})

headers = list(data.columns)
print(headers)

In this example, we have a DataFrame data with several columns. The columns method of pandas returns an Index object containing the column headers. By converting this Index object into a list using the list() function, we get a list containing the headers of the DataFrame columns. Then, we can simply use print(headers) to display the headers to the output.

Running the above code will yield an output like this:

['Column1', 'Column2', 'Column3']

This allows you to print only the headers of the DataFrame without displaying the data itself.

Displaying Results Without Truncating Column Content

To display the results without truncating the columns, you can use the pd.set_option function of pandas to configure the display of data. Here’s the modified function:

import pandas as pd
from colorama import Fore, Style

def get_mutualfund_holders(ticker):
    # Configuration for displaying data without truncating columns
    pd.set_option('display.max_colwidth', None)
    
    print()
    print(f"{Fore.LIGHTYELLOW_EX}{'FUNDS'}{Style.RESET_ALL}")
    print(ticker.mutualfund_holders)
    return

With this modification, the “Holder” column will no longer be truncated when displaying the DataFrame returned by the get_mutualfund_holders function.

List of possible display options in pandas pd.describe_option() lists and describes all possible options.

Displaying the Type of a DataFrame Column

Certainly! To identify the data type in the “Value” column of the DataFrame, you can use the dtype method of pandas on that column. Here’s how you can do it:

import pandas as pd
from colorama import Fore, Style

def get_mutualfund_holders(ticker):
    # Configuration for displaying data without truncating columns
    pd.set_option('display.max_colwidth', None)
    
    # Display the data type in the "Value" column
    print()
    print(f"{Fore.LIGHTYELLOW_EX}{'FUNDS'}{Style.RESET_ALL}")
    print("Data type of 'Value' column:", ticker.mutualfund_holders['Value'].dtype)
    
    print()
    print(ticker.mutualfund_holders)
    return

With this modification, the get_mutualfund_holders function will display the data type in the “Value” column. Make sure that the data in this column are indeed integers for the formatting with thousands separator to work correctly.