在Django的Serializer的列表数据中剔除指定元素

news2025/1/12 7:23:23

【Python工作随笔】

提问

如何在List序列化方法中剔除不要的元素，例如在成绩中剔除0

class BasicDescriptionSubjectBoxPlotSerializer(serializers.Serializer):
    语文 = serializers.ListField(source='score_chinese')
    数学 = serializers.ListField(source='score_math')
    英语 = serializers.ListField(source='score_english')
    物理 = serializers.ListField(source='score_physics')
    化学 = serializers.ListField(source='score_chemistry')
    生物 = serializers.ListField(source='score_biology')
    历史 = serializers.ListField(source='score_history')
    地理 = serializers.ListField(source='score_geography')
    政治 = serializers.ListField(source='score_politics')

解答

要在序列化过程中剔除不需要的元素，如在成绩列表中剔除分数为0的成绩，你可以通过自定义 SerializerMethodField 来处理这些数据。这样做可以在序列化过程中对数据进行更灵活的处理。

以下是如何修改 BasicDescriptionSubjectBoxPlotSerializer 序列化器来实现这一点：

from rest_framework import serializers

class BasicDescriptionSubjectBoxPlotSerializer(serializers.Serializer):
    语文 = serializers.SerializerMethodField()
    数学 = serializers.SerializerMethodField()
    英语 = serializers.SerializerMethodField()
    物理 = serializers.SerializerMethodField()
    化学 = serializers.SerializerMethodField()
    生物 = serializers.SerializerMethodField()
    历史 = serializers.SerializerMethodField()
    地理 = serializers.SerializerMethodField()
    政治 = serializers.SerializerMethodField()

    def get_语文(self, obj):
        return [score for score in obj.score_chinese if score != 0]

    def get_数学(self, obj):
        return [score for score in obj.score_math if score != 0]

    def get_英语(self, obj):
        return [score for score in obj.score_english if score != 0]

    def get_物理(self, obj):
        return [score for score in obj.score_physics if score != 0]

    def get_化学(self, obj):
        return [score for score in obj.score_chemistry if score != 0]

    def get_生物(self, obj):
        return [score for score in obj.score_biology if score != 0]

    def get_历史(self, obj):
        return [score for score in obj.score_history if score != 0]

    def get_地理(self, obj):
        return [score for score in obj.score_geography if score != 0]

    def get_政治(self, obj):
        return [score for score in obj.score_politics if score != 0]

这里 SerializerMethodField 允许定义一个方法，该方法返回你希望包含在序列化后的数据中的值。在每个方法中，我们使用列表推导来过滤掉等于0的成绩。这样，任何为0的成绩都不会被包含在最终的序列化数据中。

需要确保序列化器的实例（obj）有正确的属性，比如 score_chinese、score_math 等，这些属性应该是包含成绩的列表。如果你的数据模型或结构与此示例不同，请相应地调整属性名。

还有一种简单直接重写 to_representation 方法。

为了在序列化过程中将数据列表中的0剔除，你可以重写to_representation方法。在这个方法中，你可以遍历每个字段的数据列表，并将0过滤掉。以下是一个示例实现：

from rest_framework import serializers

class BasicDescriptionSubjectBoxPlotSerializer(serializers.Serializer):
    语文 = serializers.ListField(source='score_chinese')
    数学 = serializers.ListField(source='score_math')
    英语 = serializers.ListField(source='score_english')
    物理 = serializers.ListField(source='score_physics')
    化学 = serializers.ListField(source='score_chemistry')
    生物 = serializers.ListField(source='score_biology')
    历史 = serializers.ListField(source='score_history')
    地理 = serializers.ListField(source='score_geography')
    政治 = serializers.ListField(source='score_politics')

    def to_representation(self, instance):
        representation = super().to_representation(instance)
        for subject in representation:
            representation[subject] = [score for score in representation[subject] if score != 0]
        return representation