[英]Remove from list where all items not included using LINQ
我有一個通用列表ItemList
,如果每個ListID
都不包含所有可能的ItemIDs
,那么我想將它們從列表中刪除。
我將在下面用數據的簡化版本進行說明:
MaxItemID = 4
ListID ItemID
1 1
2 1
2 2
2 3
2 4
3 1
3 2
4 1
4 2
4 3
4 4
5 1
5 2
我想要的數據是:
ListID ItemID
2 1
2 2
2 3
2 4
4 1
4 2
4 3
4 4
昨天我問了一個僅與SQL有關的類似問題,並得到了使用Exists
的答案,這很好,但后來我發現最大項數可以變化,而且我認為我更喜歡使用linq。 我以為可以使用distinct().Count()
技術,但到目前為止沒有成功,而且聽起來也不是最有效的方法。
非常感謝
希望這是您的目的:
var list =
ItemList
.GroupBy(p=>p.ListID)
.Where(p=>p.Distinct().Count()<MaxItemID)
.ToArray()
int distinctIDs = items.Select(x => x.ItemID).Distinct().Count();
var newList = items
.GroupBy(x => x.ListID)
.Where( x => x.Select(y=>y.ItemID).Distinct().Count() == distinctIDs)
.SelectMany(x => x.ToList())
.ToList();
嘗試這個
var query = from baseItem in itemList
group baseItem by baseItem.ListID into gr
where gr.Count() == MaxItemID
join selectItem in itemList on gr.Key equals selectItem.ListID
select selectItem;
var requiredList = query.ToList();
這有效,請檢查屏幕快照。
編輯
如果允許重復,則可以將查詢修改為
var query = from item in itemList
group item by item.ListID into gr
where gr.Select(s=>s.ItemId).Distinct().Count() == MaxItemID
join i in itemList on gr.Key equals i.ListID
select i;
無法評論Linq版本的性能,但這里有Linq和非Linq解決方案
心連心,
艾倫
編輯
示例數據的確顯示了listId / ItemId的所有重復項,因此我省略了非重復檢查,但是如果需要,它將作為linq語句的第一部分。
編輯2
已更新Linq代碼。
尚未更新非linq代碼。 只需檢查一下是否與先前相同即可,然后再添加(假設按示例輸入進行了排序,ListId,ItemId)。
[TestMethod]
public void LinqVersion() {
Dictionary<int, List<ListItem>> found = new Dictionary<int, List<ListItem>>();
var comparer = new ListItemEqualityComparer();
var actual = Input.Distinct(comparer).GroupBy(lstItem => lstItem.ListId).Where(grp => grp.Count() == 4).SelectMany(grp => grp);
Assert.IsTrue(Expected.SequenceEqual(actual, comparer));
}
[TestMethod]
public void NoLinqVersion() {
Dictionary<int, List<ListItem>> found = new Dictionary<int, List<ListItem>>();
foreach (var listItem in Input) {
AddItem(found, listItem.ListId, listItem);
}
var actual = new List<ListItem>();
foreach (var pair in found) {
if (pair.Value.Count == 4) {
actual.AddRange(pair.Value);
}
}
Assert.IsTrue(Expected.SequenceEqual(actual, new ListItemEqualityComparer()));
}
private static void AddItem(IDictionary<int, List<ListItem>> dictionary, int listId, ListItem listItem) {
if (!dictionary.ContainsKey(listId)) {
dictionary.Add(listId, new List<ListItem>());
}
dictionary[listId].Add(listItem);
}
public class ListItem {
public int ListId { get; set; }
public int ItemId { get; set; }
}
public class ListItemEqualityComparer : IEqualityComparer<ListItem> {
public bool Equals(ListItem x, ListItem y) {
return x.ListId == y.ListId && x.ItemId == y.ItemId;
}
public int GetHashCode(ListItem obj) {
return obj.ListId ^ obj.ItemId;
}
}
public List<ListItem> Input = new List<ListItem>(){
new ListItem{ ListId = 1 , ItemId = 1},
new ListItem{ ListId = 2 , ItemId = 1},
new ListItem{ ListId = 2 , ItemId = 2},
new ListItem{ ListId = 2 , ItemId = 3},
new ListItem{ ListId = 2 , ItemId = 4},
new ListItem{ ListId = 3 , ItemId = 1},
new ListItem{ ListId = 3 , ItemId = 2},
new ListItem{ ListId = 4 , ItemId = 1},
new ListItem{ ListId = 4 , ItemId = 2},
new ListItem{ ListId = 4 , ItemId = 3},
new ListItem{ ListId = 4 , ItemId = 4},
new ListItem{ ListId = 5 , ItemId = 1},
new ListItem{ ListId = 5 , ItemId = 2},
};
public List<ListItem> Expected = new List<ListItem>{
new ListItem{ ListId = 2 , ItemId = 1},
new ListItem{ ListId = 2 , ItemId = 2},
new ListItem{ ListId = 2 , ItemId = 3},
new ListItem{ ListId = 2 , ItemId = 4},
new ListItem{ ListId = 4 , ItemId = 1},
new ListItem{ ListId = 4 , ItemId = 2},
new ListItem{ ListId = 4 , ItemId = 3},
new ListItem{ ListId = 4 , ItemId = 4}
};
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.