import pandas as pd
import pandas as pd

# 显示所有列，不省略
pd.set_option('display.max_columns', None)

# 显示所有行（对 head() 没影响，但可用于 df 全体显示时）
pd.set_option('display.max_rows', None)

# 不截断列内容，显示完整字符串
pd.set_option('display.max_colwidth', None)

# 设置最大宽度，防止自动换行
pd.set_option('display.width', 1000)
# 读取 parquet 文件（自动识别使用的 engine，比如 pyarrow 或 fastparquet）
df = pd.read_parquet("/workspace/0525_zyw/verl/counting/mk_data/v2_train_counting_dataset_OpenR1-Math-220k_90258.parquet", engine="auto")

print(df.shape)
# print(df.columns)
# # 查看前几条数据
# print(df.loc[0])  # 默认前 5 条，可以传入参数指定行数，比如 df.head(10)
# for i, row in df.iterrows():
#     print(f"Row {i}:\n{row['prompt']}\n---")
#     if i > 1:  # 只打印前 5 行
#         break