LINQを使ったデータテーブルからの抽出について

Question

開発環境

Windows10
VisualStudio 2017
VB.NET
.Net Framework 4.6.2

前提・実現したいこと

LINQを使い、データテーブルから条件に合致するデータを抽出したいと考えています。
例として下のテストデータを例に詳細を書きます。

KEY(STRING)	CODE1(STRING)	CODE2(STRING)	VALUE(STRING)
1	001	0	テスト１-１
2	001	1	テスト１-２
3	002	0	テスト２-１
4	002	1	テスト２-２
5	003	0	テスト３-１
6	003	1	テスト３-２

上記のようなデータから、CODE1とCODE2を条件にデータの抽出を行います。
具体的な抽出条件は下記のとおりです。

・CODE1が「001」の場合：
CODE1が「001」のデータを取得

・CODE1が「002」の場合：
CODE1が「002」かつ、CODE2が「0」のデータを取得

・CODE1が「003」の場合：
CODE1が「003」かつ、CODE2が「0以上」のデータを取得

プログラム

'検索条件となるCODE1
'searchConditionの中身は動的に変わることを想定
'"001"だけのこともあれば"001,002,003"の場合もある
Dim searchCondition() As String = New String() {"001", "003"}
Dim dt As Datatable = 前提・実現したいことに記載したデータ

Dim linq = From row In dt.AsEnumerable
           Where 1 = 1

For Each str As String In searchCondition
    Select Case str
        Case "001"
            linq = linq.Where(Function(row) row("CODE1") = str)
        Case "002"
            linq = linq.Where(Function(row) row("CODE1") = str And row("CODE2") = 0)
        Case "003"
            linq = linq.Where(Function(row) row("CODE1") = str And row("CODE2") >= 0)
Next

期待値・実行結果

■期待値
KEY=1、KEY=2、KEY=6のデータを抽出したい
(row("CODE1") = str Or (linq.Where(Function(row) row("CODE1") = str And row("CODE2") >= 0))

■実行結果
CODE1=001のWhere句とCODE1=003のWhere句がAnd条件となってしまい抽出結果0件
(row("CODE1") = str And (linq.Where(Function(row) row("CODE1") = str And row("CODE2") >= 0))

複数のWhere句が作られた場合に、それらをOr条件で実行したいと考えています。

Comments

[退会済みユーザー]

開発環境ぐらいは書きましょう

[退会済みユーザー]

VB.NET のタグをつけてください。

DataTable の各列の型を質問欄を編集して追記してください。

質問者さんの書いたコードの実行結果はどうなるのか、どこがどのように期待と異なるのかを質問欄を編集して追記してください。

> CODE1が「003」かつ、CODE2が「0以上」のデータを取得

だけが問題で、その ling だけ分かれば解決するように見えますが、そうなんですか？ ・・・と思ったけど違うみたいですね。

Or でつなぐ？ どういうことか、期待する結果を具体的に書くなどして説明できませんか？

[退会済みユーザー]

抽出条件は searchCondition の文字列の配列で、その組み合わせは以下の 7 通りあって、

"001"
"002"
"003"
"001", "002"
"001", "003"
"002", "003"
"001", "002", "003"

例えば "001", "003" の場合は、

CODE1が「001」のデータ

と

CODE1が「003」かつ、CODE2が「0以上」のデータ

を linq に取得するということですか？

[conn_e]

はい、そうです

[退会済みユーザー]

List<DataRow> 型にして良いなら List<T>.AddRange メソッドを使って追加していくという方法がありますが、それでいいですか？

[退会済みユーザー]

List<DataRow> 型にしなくても（EnumerableRowCollection<DataRow> 型のままでも）Concat を使って可能でした。後でその案を回答欄に書いておきます。

Answer 1

抽出条件は searchCondition の文字列の配列で、その組み合わせは以下の 7 通りあって、

"001"
"002"
"003"
"001", "002"
"001", "003"
"002", "003"
"001", "002", "003"

そして、例えば searchCondition の中身が "001", "003" の場合は、

(1) CODE1が「001」のデータ

と

(2) CODE1が「003」かつ、CODE2が「0以上」のデータ

をまとめて EnumerableRowCollection<DataRow> オブジェクトとして取得したいということと理解してレスします。

以下のように Concat メソッドを使って上の (1) と (2) の結果を連結すればできると思います。コードは C# ですので注意してください。

// CODE1が「001」の場合： CODE1が「001」のデータを取得
var linq = from row in dt.AsEnumerable()
           where row.Field<string>("CODE1") == "001"
           select row;

// CODE1が「003」の場合： CODE1が「003」かつ、CODE2が「0以上」のデータを取得
var linq2 = from row in dt.AsEnumerable()
            where row.Field<string>("CODE1") == "003" &&
            row.Field<string>("CODE2").CompareTo("0") > 0
            select row;

var linq3 = linq.Concat(linq2);

foreach (var x in linq3)
{
    Console.WriteLine($"{x["KEY"]}, {x["CODE1"]}, {x["CODE2"]}, {x["VALUE"]}");
}

結果は：

1, 001, 0, テスト１-１
2, 001, 1, テスト１-２
6, 003, 1, テスト３-２

7 通り全部にどう対応するかは質問者さんの方で考えてください。