abc

test.py

@@ -0,0 +1,37 @@
+import csv
+import pandas as pd
+import _pickle as pkl
+
+
+def trans_csv_to_pickle(filename):
+    df = pd.read_csv(filename)
+    save_df = pd.DataFrame(columns=["features", "label"])
+    d_index = list(df.columns).index('features')
+    print(df)
+    for i in range(len(df)):
+        features = df.loc[i].features.replace("inf", str(100000000.0))
+        try:
+            features = eval(features)
+        except NameError:
+            print(features)
+            inf = 1e10
+            eval(features)
+            print(features)
+            continue
+        for feature in features:
+            if not isinstance(feature, float):
+                print(features)
+                print(feature)
+        # print(df.loc[i].label, type(df.loc[i].label))
+        save_df.loc[i] = [features, int(df.loc[i].label)]
+    print(save_df)
+    print(type(save_df.loc[0]['features']))
+    save_df.to_csv(filename)
+    pkl_name = filename.replace("csv", "pkl")
+    f_pkl = open(pkl_name, "wb")
+    pkl.dump(save_df, f_pkl)
+    f_pkl.close()
+
+
+if __name__ == '__main__':
+    trans_csv_to_pickle("result/ow_doh_features.csv")