如何使用linq删除项目

Question

我有以下数组,它表示我试图从某个数据结构实例中删除的项目:

string[] diseasesToRemove = new string[] { "D1", "D3", "D5" };

我可以使用以下方法完成:

for (int i = 0; i < healthGroup.DiseaseGroups.Length; i++)
{
    var dgDiseases = new List<Disease>(healthGroup.DiseaseGroups[i].Diseases);
    for (int j = 0; j < dgDiseases.Count; j++)
    {
        if (diseasesToRemove.Contains(dgDiseases[j].Name))
        {                 
            dgDiseases.RemoveAt(j);
            j--;
        }
    }
    healthGroup.DiseaseGroups[i].Diseases = dgDiseases.ToArray();
}

但是,我确信使用Linq或其他方法有更好的方法.在那儿？

以下是课程:

public class HealthGroup
{
    public DiseaseGroup[] DiseaseGroups { get; set; }

    public HealthGroup(DiseaseGroup[] diseaseGroups)
    {
        DiseaseGroups = diseaseGroups;
    }
}

public class DiseaseGroup
{
    public string Name { get; set; }
    public Disease[] Diseases;

    public DiseaseGroup(string name, Disease[] diseases)
    {
        Name = name;
        Diseases = diseases;
    }
}

public class Disease
{
    public string Name { get; set; } = "My Disease";
    public int Risk { get; set; } = 7;

    public Disease(string name, int risk)
    {
        Name = name;
        Risk = risk;
    }

    public override string ToString()
    {
        return $"{Name} with risk {Risk}";
    }
}

和一些样板生成实例:

Disease d1 = new Disease("D1", 1);
Disease d2 = new Disease("D2", 2);
Disease d3 = new Disease("D3", 3);
Disease d4 = new Disease("D4", 4);
Disease d5 = new Disease("D5", 5);
Disease d6 = new Disease("D6", 6);
Disease d7 = new Disease("D7", 7);

DiseaseGroup dg1 = new DiseaseGroup("DG1", new Disease[] { d1, d2 });
DiseaseGroup dg2 = new DiseaseGroup("DG2", new Disease[] { d3, d4, d5 });
DiseaseGroup dg3 = new DiseaseGroup("DG3", new Disease[] { d6, d7 });

HealthGroup healthGroup = new HealthGroup(new DiseaseGroup[] { dg1, dg2, dg3 });

Answer 1

您可以使用它Where来简化代码:

foreach (var diseaseGroup in healthGroup.DiseaseGroups)
{
    diseaseGroup.Diseases 
        = diseaseGroup.Diseases.Where(g => !diseasesToRemove.Contains(g.Name)).ToArray();
}

当然,与原始代码一样,这会生成一个新列表.更高性能的选项(尽管非LINQ)可能会做出Diseases一个List<Disease>:

public class DiseaseGroup
{
    public string Name { get; set; }
    public List<Disease> Diseases;

    public DiseaseGroup(string name, Disease[] diseases)
    {
        Name = name;
        Diseases = new List<Disease>(diseases);
    }
}

然后您可以使用类似的代码,而无需生成新列表和新数组的开销:

for (int i = 0; i < healthGroup.DiseaseGroups.Length; i++)
{
    for (int j = healthGroup.DiseaseGroups[i].Diseases.Count - 1; j >= 0; --j)
    {
        if (diseasesToRemove.Contains(healthGroup.DiseaseGroups[i].Diseases[j].Name))
        {                 
            healthGroup.DiseaseGroups[i].Diseases.RemoveAt(j);
        }
    }
}

我还修改了你的for循环以向后工作以解决你对该--j部件的问题.

并且,对于大量项目而不是使用string[]for diseasesToRemove,使用a HashSet<string>来存储疾病可能更好.

编辑以包括请求的单行:

healthGroup.DiseaseGroups = healthGroup.DiseaseGroups.Select(g => { return g.Diseases = g.Diseases.Where(g => !diseasesToRemove.Contains(g.Name)).ToArray(); }).ToArray();

这是滥用选择一点,但是:-)