使用python创建分类树的类

Question

我正在为我的大学使用python进行大数据项目。 我正在使用Kaggle的数据集（Difference between save and saveAndFlush in Spring data jpa）。最终目的是创建一个分类树。所以现在，我坚持如何在一个变量（例如imbd score）中创建一个类，如果变量小于70，则可以调用0；如果变量大于70，则可以调用类1。