22 lines
562 B
Python
22 lines
562 B
Python
import pandas as pd
|
||
|
||
|
||
def extract_features(file_path):
|
||
# 读取csv文件
|
||
df = pd.read_csv(file_path, delimiter=',')
|
||
|
||
# 按第2列数值降序排序
|
||
df["count"] = pd.to_numeric(df["count"], errors='coerce')
|
||
df_sorted = df.sort_values(by='count', ascending=True)
|
||
|
||
# 筛选出第2列值大于10000的行，并提取第1列内容
|
||
features = df_sorted[df_sorted['count'] <0]
|
||
|
||
return features
|
||
|
||
if __name__ == '__main__':
|
||
|
||
# 使用函数，传入csv文件路径
|
||
features = extract_features('./out/3gram.csv')
|
||
print(features)
|